From 316236f0e2a2f26dc9eccbe9db6f591965a123b0 Mon Sep 17 00:00:00 2001 From: jbrockmendel Date: Wed, 14 Sep 2022 14:31:53 -0700 Subject: [PATCH] Backport PR #48550: TST: remove 2D tests irrelevant for pyarrow --- pandas/tests/extension/base/dim2.py | 3 +++ pandas/tests/extension/test_arrow.py | 14 -------------- 2 files changed, 3 insertions(+), 14 deletions(-) diff --git a/pandas/tests/extension/base/dim2.py b/pandas/tests/extension/base/dim2.py index f71f3cf164bfc..1d5a5c4532a5d 100644 --- a/pandas/tests/extension/base/dim2.py +++ b/pandas/tests/extension/base/dim2.py @@ -12,6 +12,9 @@ class Dim2CompatTests(BaseExtensionTests): + # Note: these are ONLY for ExtensionArray subclasses that support 2D arrays. + # i.e. not for pyarrow-backed EAs. + def test_transpose(self, data): arr2d = data.repeat(2).reshape(-1, 2) shape = arr2d.shape diff --git a/pandas/tests/extension/test_arrow.py b/pandas/tests/extension/test_arrow.py index ce30e3e92a4c9..62e9503286311 100644 --- a/pandas/tests/extension/test_arrow.py +++ b/pandas/tests/extension/test_arrow.py @@ -319,20 +319,6 @@ def test_from_sequence_of_strings_pa_array(self, data, request): tm.assert_extension_array_equal(result, data) -@pytest.mark.xfail( - raises=NotImplementedError, reason="pyarrow.ChunkedArray backing is 1D." -) -class TestDim2Compat(base.Dim2CompatTests): - pass - - -@pytest.mark.xfail( - raises=NotImplementedError, reason="pyarrow.ChunkedArray backing is 1D." -) -class TestNDArrayBacked2D(base.NDArrayBacked2DTests): - pass - - class TestGetitemTests(base.BaseGetitemTests): @pytest.mark.xfail( reason=(