This commit is contained in:
Phillip Cloud committed 2017-11-10 17:39:57 -05:00
1 parent 9abad95fb4
commit d61c43e68f
1 file changed
+23
+23
View File
@@ -1458,3 +1458,26 @@ def test_index_column_name_duplicate(tmpdir):
arrow_table = _read_table(path)
result_df = arrow_table.to_pandas()
tm.assert_frame_equal(result_df, dfx)
def test_backwards_compatible_index_naming():
expected_string = """\
carat cut color clarity depth table price x y z
0.23 Ideal E SI2 61.5 55.0 326 3.95 3.98 2.43
0.21 Premium E SI1 59.8 61.0 326 3.89 3.84 2.31
0.23 Good E VS1 56.9 65.0 327 4.05 4.07 2.31
0.29 Premium I VS2 62.4 58.0 334 4.20 4.23 2.63
0.31 Good J SI2 63.3 58.0 335 4.34 4.35 2.75
0.24 Very Good J VVS2 62.8 57.0 336 3.94 3.96 2.48
0.24 Very Good I VVS1 62.3 57.0 336 3.95 3.98 2.47
0.26 Very Good H SI1 61.9 55.0 337 4.07 4.11 2.53
0.22 Fair E VS2 65.1 61.0 337 3.87 3.78 2.49
0.23 Very Good H VS1 59.4 61.0 338 4.00 4.05 2.39"""
expected = pd.read_csv(
io.StringIO(expected_string), sep=r'\s{2,}', index_col=None,
header=0
)
path = os.path.join(os.path.dirname(__file__), 'data', 'v0.7.1.parquet')
t = _read_table(path)
result = t.to_pandas()
tm.assert_frame_equal(result, expected)