Skip to content

Commit c22bc65

Browse files
authored
CI/TST: Fix test for pyarrow 8.0 release (#46991)
1 parent 04e01a1 commit c22bc65

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

pandas/compat/pyarrow.py

+2
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
pa_version_under5p0 = _palv < Version("5.0.0")
1515
pa_version_under6p0 = _palv < Version("6.0.0")
1616
pa_version_under7p0 = _palv < Version("7.0.0")
17+
pa_version_under8p0 = _palv < Version("8.0.0")
1718
except ImportError:
1819
pa_version_under1p01 = True
1920
pa_version_under2p0 = True
@@ -22,3 +23,4 @@
2223
pa_version_under5p0 = True
2324
pa_version_under6p0 = True
2425
pa_version_under7p0 = True
26+
pa_version_under8p0 = True

pandas/tests/io/test_parquet.py

+7-3
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
pa_version_under2p0,
1919
pa_version_under5p0,
2020
pa_version_under6p0,
21+
pa_version_under8p0,
2122
)
2223
import pandas.util._test_decorators as td
2324

@@ -718,11 +719,14 @@ def test_duplicate_columns(self, pa):
718719
df = pd.DataFrame(np.arange(12).reshape(4, 3), columns=list("aaa")).copy()
719720
self.check_error_on_write(df, pa, ValueError, "Duplicate column names found")
720721

721-
def test_unsupported(self, pa):
722-
# timedelta
722+
def test_timedelta(self, pa):
723723
df = pd.DataFrame({"a": pd.timedelta_range("1 day", periods=3)})
724-
self.check_external_error_on_write(df, pa, NotImplementedError)
724+
if pa_version_under8p0:
725+
self.check_external_error_on_write(df, pa, NotImplementedError)
726+
else:
727+
check_round_trip(df, pa)
725728

729+
def test_unsupported(self, pa):
726730
# mixed python objects
727731
df = pd.DataFrame({"a": ["a", 1, 2.0]})
728732
# pyarrow 0.11 raises ArrowTypeError

0 commit comments

Comments
 (0)