modified: pandas/tests/reshape/test_pivot_multilevel.py

	- Moved e_data, e_index, and e_cols to parametrized marker instead of declaring inside the test :func:`test_pivot_table_multiindex_values_as_two_params`.

Author: it176131

pandas/tests/reshape/test_pivot_multilevel.py

@@ -253,11 +253,41 @@ def test_pivot_df_multiindex_index_none():
 
 
 @pytest.mark.parametrize(
-    "index, columns",
-    [("index", ["col", "value"]), (["index", "value"], "col")],
+    "index, columns, e_data, e_index, e_cols",
+    [
+        (
+            "index",
+            ["col", "value"],
+            [
+                [50.0, (nan := np.nan), 100.0, nan],
+                [nan, 100.0, nan, 200.0],
+            ],
+            Index(data=["A", "B"], name="index"),
+            MultiIndex.from_arrays(
+                arrays=[[1, 1, 2, 2], [50, 100, 100, 200]], names=["col", "value"]
+            ),
+        ),
+        (
+            ["index", "value"],
+            "col",
+            [
+                [50.0, nan],
+                [nan, 100.0],
+                [100.0, nan],
+                [nan, 200.0],
+            ],
+            MultiIndex.from_arrays(
+                arrays=[["A", "A", "B", "B"], [50, 100, 100, 200]],
+                names=["index", "value"],
+            ),
+            Index(data=[1, 2], name="col"),
+        ),
+    ],
     ids=["values-and-columns", "values-and-index"],
 )
-def test_pivot_table_multiindex_values_as_two_params(index, columns, request):
+def test_pivot_table_multiindex_values_as_two_params(
+    index, columns, e_data, e_index, e_cols
+):
     # GH#61292
     data = [
         ["A", 1, 50, -1],
@@ -267,28 +297,5 @@ def test_pivot_table_multiindex_values_as_two_params(index, columns, request):
     ]
     df = pd.DataFrame(data=data, columns=["index", "col", "value", "extra"])
     result = df.pivot_table(values="value", index=index, columns=columns)
-    nan = np.nan
-    if request.node.callspec.id == "values-and-columns":
-        e_data = [
-            [50.0, nan, 100.0, nan],
-            [nan, 100.0, nan, 200.0],
-        ]
-        e_index = Index(data=["A", "B"], name="index")
-        e_cols = MultiIndex.from_arrays(
-            arrays=[[1, 1, 2, 2], [50, 100, 100, 200]], names=["col", "value"]
-        )
-
-    else:
-        e_data = [
-            [50.0, nan],
-            [nan, 100.0],
-            [100.0, nan],
-            [nan, 200.0],
-        ]
-        e_index = MultiIndex.from_arrays(
-            arrays=[["A", "A", "B", "B"], [50, 100, 100, 200]], names=["index", "value"]
-        )
-        e_cols = Index(data=[1, 2], name="col")
-
     expected = pd.DataFrame(data=e_data, index=e_index, columns=e_cols)
     tm.assert_frame_equal(result, expected)