test: improve test parametrization

Author: Alvaro-Kothe

pandas/tests/frame/test_stack_unstack.py

@@ -1387,37 +1387,34 @@ def test_unstack_sort_false(frame_or_series, dtype):
 
 
 @pytest.mark.parametrize(
-    "levels2, expected_columns, expected_data",
+    "levels2, expected_columns",
     [
         (
-            Index([None, 1, 2, 3]),
-            [("value", np.nan), ("value", 1.0), ("value", 2.0), ("value", 3.0)],
-            [[0, 4], [1, 5], [2, 6], [3, 7]],
+            [None, 1, 2, 3],
+            [("value", np.nan), ("value", 1), ("value", 2), ("value", 3)],
         ),
         (
-            Index([1, None, 2, 3]),
-            [("value", 1.0), ("value", np.nan), ("value", 2.0), ("value", 3.0)],
-            [[0, 4], [1, 5], [2, 6], [3, 7]],
+            [1, None, 2, 3],
+            [("value", 1), ("value", np.nan), ("value", 2), ("value", 3)],
         ),
         (
-            Index([1, 2, None, 3]),
-            [("value", 1.0), ("value", 2.0), ("value", np.nan), ("value", 3.0)],
-            [[0, 4], [1, 5], [2, 6], [3, 7]],
+            [1, 2, None, 3],
+            [("value", 1), ("value", 2), ("value", np.nan), ("value", 3)],
         ),
         (
-            Index([1, 2, 3, None]),
-            [("value", 1.0), ("value", 2.0), ("value", 3.0), ("value", np.nan)],
-            [[0, 4], [1, 5], [2, 6], [3, 7]],
+            [1, 2, 3, None],
+            [("value", 1), ("value", 2), ("value", 3), ("value", np.nan)],
         ),
     ],
     ids=["nan=first", "nan=second", "nan=third", "nan=last"],
 )
-def test_unstack_sort_false_nan(levels2, expected_columns, expected_data):
+def test_unstack_sort_false_nan(levels2, expected_columns):
     # GH#61221
     levels1 = ["b", "a"]
     index = MultiIndex.from_product([levels1, levels2], names=["level1", "level2"])
     df = DataFrame({"value": [0, 1, 2, 3, 4, 5, 6, 7]}, index=index)
     result = df.unstack(level="level2", sort=False)
+    expected_data = [[0, 4], [1, 5], [2, 6], [3, 7]]
     expected = DataFrame(
         dict(zip(expected_columns, expected_data)),
         index=Index(["b", "a"], name="level1"),