Skip to content

Commit cd99b01

Browse files
Backport PR #46991: CI/TST: Fix test for pyarrow 8.0 release (#46992)
Co-authored-by: Matthew Roeschke <[email protected]>
1 parent b18925d commit cd99b01

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
@@ -17,6 +17,7 @@
1717
pa_version_under2p0,
1818
pa_version_under5p0,
1919
pa_version_under6p0,
20+
pa_version_under8p0,
2021
)
2122
import pandas.util._test_decorators as td
2223

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

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

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

0 commit comments

Comments
 (0)