CLN: Fix issues reported by pre-commit hooks

Author: robin-mader-bis

pandas/tests/reshape/test_pivot.py

@@ -2856,9 +2856,14 @@ def test_pivot_margins_with_none_index(self):
     def test_pivot_with_pyarrow_categorical(self):
         # GH#53051
 
-        # Create dataframe with categorical colum
+        pytest.importorskip("pyarrow")
+
+        # Create dataframe with categorical column
         df = (
-            pd.DataFrame([("A", 1), ("B", 2), ("C", 3)], columns=["string_column", "number_column"])
+            DataFrame(
+                [("A", 1), ("B", 2), ("C", 3)],
+                columns=["string_column", "number_column"],
+            )
             .astype({"string_column": "string", "number_column": "float32"})
             .astype({"string_column": "category", "number_column": "float32"})
         )
@@ -2869,25 +2874,18 @@ def test_pivot_with_pyarrow_categorical(self):
             buffer.seek(0)  # Reset buffer position
             df = pd.read_parquet(buffer, dtype_backend="pyarrow")
 
-
         # Check that pivot works
         df = df.pivot(columns=["string_column"], values=["number_column"])
 
         # Assert that values of result are correct to prevent silent failure
-        multi_index = pd.MultiIndex.from_arrays(
-            [
-                ["number_column", "number_column", "number_column"],
-                ["A", "B", "C"]
-            ],
-            names=(None, "string_column")
+        multi_index = MultiIndex.from_arrays(
+            [["number_column", "number_column", "number_column"], ["A", "B", "C"]],
+            names=(None, "string_column"),
         )
-        df_expected = pd.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(
+            [[1.0, np.nan, np.nan], [np.nan, 2.0, np.nan], [np.nan, np.nan, 3.0]],
+            columns=multi_index,
+        )
+        tm.assert_frame_equal(
+            df, df_expected, check_dtype=False, check_column_type=False
         )
-        tm.assert_frame_equal(df, df_expected, check_dtype=False, check_column_type=False)
-

pandas/tests/test_multilevel.py

@@ -322,9 +322,14 @@ def test_multiindex_dt_with_nan(self):
     def test_multiindex_with_pyarrow_categorical(self):
         # GH#53051
 
-        # Create dataframe with categorical colum
+        pytest.importorskip("pyarrow")
+
+        # Create dataframe with categorical column
         df = (
-            pd.DataFrame([("A", 1), ("B", 2), ("C", 3)], columns=["string_column", "number_column"])
+            DataFrame(
+                [["A", 1], ["B", 2], ["C", 3]],
+                columns=["string_column", "number_column"],
+            )
             .astype({"string_column": "string", "number_column": "float32"})
             .astype({"string_column": "category", "number_column": "float32"})
         )
@@ -335,7 +340,6 @@ def test_multiindex_with_pyarrow_categorical(self):
             buffer.seek(0)  # Reset buffer position
             df = pd.read_parquet(buffer, dtype_backend="pyarrow")
 
-
         # Check that index can be set
         df.set_index(["string_column", "number_column"])