fix on pyarrow-less builds

jbrockmendel · Aug 1, 2023 · c5156eb · c5156eb
1 parent 360eb9d
commit c5156eb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pandas/tests/extension/base/methods.py b/pandas/tests/extension/base/methods.py
@@ -66,7 +66,9 @@ def test_value_counts_with_normalize(self, data):
             expected = pd.Series(0.0, index=result.index, name="proportion")
             expected[result > 0] = 1 / len(values)
 
-        if data.dtype == "string[pyarrow]" or isinstance(data.dtype, pd.ArrowDtype):
+        if getattr(data.dtype, "storage", "") == "pyarrow" or isinstance(
+            data.dtype, pd.ArrowDtype
+        ):
             # TODO: avoid special-casing
             expected = expected.astype("double[pyarrow]")
         elif na_value_for_dtype(data.dtype) is pd.NA: