From f51c2a03ba0dab0431182db366e95af70aadb2bd Mon Sep 17 00:00:00 2001 From: MeeseeksMachine <39504233+meeseeksmachine@users.noreply.github.com> Date: Thu, 15 Sep 2022 02:04:43 +0200 Subject: [PATCH] Backport PR #48550 on branch 1.5.x (TST: remove 2D tests irrelevant for pyarrow) (#48554) Backport PR #48550: TST: remove 2D tests irrelevant for pyarrow Co-authored-by: jbrockmendel --- 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=(