From 26c0fea864089e2e784ca45183b43ec50f021401 Mon Sep 17 00:00:00 2001 From: Matthew Roeschke Date: Tue, 10 May 2022 15:29:29 -0700 Subject: [PATCH] CI/TST: Fix test for pyarrow 8.0 release --- pandas/compat/pyarrow.py | 2 ++ pandas/tests/io/test_parquet.py | 10 +++++++--- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/pandas/compat/pyarrow.py b/pandas/compat/pyarrow.py index 00b205d018e89..eef2bb6639c36 100644 --- a/pandas/compat/pyarrow.py +++ b/pandas/compat/pyarrow.py @@ -14,6 +14,7 @@ pa_version_under5p0 = _palv < Version("5.0.0") pa_version_under6p0 = _palv < Version("6.0.0") pa_version_under7p0 = _palv < Version("7.0.0") + pa_version_under8p0 = _palv < Version("8.0.0") except ImportError: pa_version_under1p01 = True pa_version_under2p0 = True @@ -22,3 +23,4 @@ pa_version_under5p0 = True pa_version_under6p0 = True pa_version_under7p0 = True + pa_version_under8p0 = True diff --git a/pandas/tests/io/test_parquet.py b/pandas/tests/io/test_parquet.py index 7c04a51e803f6..5b899079dfffd 100644 --- a/pandas/tests/io/test_parquet.py +++ b/pandas/tests/io/test_parquet.py @@ -18,6 +18,7 @@ pa_version_under2p0, pa_version_under5p0, pa_version_under6p0, + pa_version_under8p0, ) import pandas.util._test_decorators as td @@ -718,11 +719,14 @@ def test_duplicate_columns(self, pa): df = pd.DataFrame(np.arange(12).reshape(4, 3), columns=list("aaa")).copy() self.check_error_on_write(df, pa, ValueError, "Duplicate column names found") - def test_unsupported(self, pa): - # timedelta + def test_timedelta(self, pa): df = pd.DataFrame({"a": pd.timedelta_range("1 day", periods=3)}) - self.check_external_error_on_write(df, pa, NotImplementedError) + if pa_version_under8p0: + self.check_external_error_on_write(df, pa, NotImplementedError) + else: + check_round_trip(df, pa) + def test_unsupported(self, pa): # mixed python objects df = pd.DataFrame({"a": ["a", 1, 2.0]}) # pyarrow 0.11 raises ArrowTypeError