|
1 | 1 | """ test parquet compat """
|
2 | 2 | import datetime
|
| 3 | +from distutils.version import LooseVersion |
3 | 4 | import os
|
4 | 5 | from warnings import catch_warnings
|
5 | 6 |
|
@@ -166,6 +167,7 @@ def compare(repeat):
|
166 | 167 | df.to_parquet(path, **write_kwargs)
|
167 | 168 | with catch_warnings(record=True):
|
168 | 169 | actual = read_parquet(path, **read_kwargs)
|
| 170 | + |
169 | 171 | tm.assert_frame_equal(expected, actual, check_names=check_names)
|
170 | 172 |
|
171 | 173 | if path is None:
|
@@ -453,9 +455,12 @@ def test_categorical(self, pa):
|
453 | 455 | # supported in >= 0.7.0
|
454 | 456 | df = pd.DataFrame({"a": pd.Categorical(list("abc"))})
|
455 | 457 |
|
456 |
| - # de-serialized as object |
457 |
| - expected = df.assign(a=df.a.astype(object)) |
458 |
| - check_round_trip(df, pa, expected=expected) |
| 458 | + if LooseVersion(pyarrow.__version__) >= LooseVersion("0.15.0"): |
| 459 | + check_round_trip(df, pa) |
| 460 | + else: |
| 461 | + # de-serialized as object for pyarrow < 0.15 |
| 462 | + expected = df.assign(a=df.a.astype(object)) |
| 463 | + check_round_trip(df, pa, expected=expected) |
459 | 464 |
|
460 | 465 | def test_s3_roundtrip(self, df_compat, s3_resource, pa):
|
461 | 466 | # GH #19134
|
|
0 commit comments