paramertize test for df.convert_dtypes()

Author: simonjayhawkins

pandas/tests/frame/methods/test_convert_dtypes.py

@@ -1,6 +1,8 @@
 import numpy as np
 import pytest
 
+import pandas.util._test_decorators as td
+
 import pandas as pd
 import pandas._testing as tm
 
@@ -9,7 +11,16 @@ class TestConvertDtypes:
     @pytest.mark.parametrize(
         "convert_integer, expected", [(False, np.dtype("int32")), (True, "Int32")]
     )
-    def test_convert_dtypes(self, convert_integer, expected):
+    @pytest.mark.parametrize(
+        "string_storage",
+        [
+            "python",
+            pytest.param(
+                "pyarrow", marks=td.skip_if_no("pyarrow", min_version="1.0.0")
+            ),
+        ],
+    )
+    def test_convert_dtypes(self, convert_integer, expected, string_storage):
         # Specific types are tested in tests/series/test_dtypes.py
         # Just check that it works for DataFrame here
         df = pd.DataFrame(
@@ -18,11 +29,12 @@ def test_convert_dtypes(self, convert_integer, expected):
                 "b": pd.Series(["x", "y", "z"], dtype=np.dtype("O")),
             }
         )
-        result = df.convert_dtypes(True, True, convert_integer, False)
+        with pd.option_context("string_storage", string_storage):
+            result = df.convert_dtypes(True, True, convert_integer, False)
         expected = pd.DataFrame(
             {
                 "a": pd.Series([1, 2, 3], dtype=expected),
-                "b": pd.Series(["x", "y", "z"], dtype="string"),
+                "b": pd.Series(["x", "y", "z"], dtype=f"string[{string_storage}]"),
             }
         )
         tm.assert_frame_equal(result, expected)