CLN: Refactor tests to adress PR feedback

Author: robin-mader-bis

pandas/tests/reshape/test_pivot.py

@@ -2858,8 +2858,9 @@ def test_pivot_with_pyarrow_categorical(self):
         # GH#53051
         pa = pytest.importorskip("pyarrow")
 
+        data = {"string_column": ["A", "B", "C"], "number_column": [1, 2, 3]}
         df = (
-            DataFrame({"string_column": ["A", "B", "C"], "number_column": [1, 2, 3]})
+            DataFrame(data)
             .astype({"string_column": "category", "number_column": "float32"})
             .astype(
                 {
@@ -2871,14 +2872,7 @@ def test_pivot_with_pyarrow_categorical(self):
 
         df = df.pivot(columns=["string_column"], values=["number_column"])
 
-        multi_index = MultiIndex.from_arrays(
-            [["number_column", "number_column", "number_column"], ["A", "B", "C"]],
-            names=(None, "string_column"),
-        )
-        df_expected = DataFrame(
-            [[1.0, np.nan, np.nan], [np.nan, 2.0, np.nan], [np.nan, np.nan, 3.0]],
-            columns=multi_index,
-        )
+        df_expected = DataFrame(data).pivot(columns=["string_column"], values=["number_column"])
         tm.assert_frame_equal(
             df, df_expected, check_dtype=False, check_column_type=False
         )

pandas/tests/test_multilevel.py

@@ -324,8 +324,9 @@ def test_multiindex_with_pyarrow_categorical(self):
         # GH#53051
         pa = pytest.importorskip("pyarrow")
 
+        data = {"string_column": ["A", "B", "C"], "number_column": [1, 2, 3]}
         df = (
-            DataFrame({"string_column": ["A", "B", "C"], "number_column": [1, 2, 3]})
+            DataFrame(data)
             .astype({"string_column": "category", "number_column": "float32"})
             .astype(
                 {
@@ -337,13 +338,9 @@ def test_multiindex_with_pyarrow_categorical(self):
 
         df = df.set_index(["string_column", "number_column"])
 
-        df_expected = DataFrame(
-            index=MultiIndex.from_arrays(
-                [["A", "B", "C"], [1, 2, 3]], names=["string_column", "number_column"]
-            )
-        )
+        df_expected = DataFrame(data).set_index(["string_column", "number_column"])
         tm.assert_frame_equal(
-            df, df_expected, check_dtype=False, check_column_type=False
+            df, df_expected, check_dtype=False, check_index_type=False,
         )