modified: pandas/tests/reshape/test_pivot_multilevel.py

it176131 · it176131 · commit 6bdb3deece36 · 2025-04-19T13:42:07.000-05:00
- Reduced `expected` assignments from two to one per feedback.
diff --git a/pandas/tests/reshape/test_pivot_multilevel.py b/pandas/tests/reshape/test_pivot_multilevel.py
@@ -277,7 +277,6 @@ def test_pivot_table_multiindex_values_as_two_params(index, columns, request):
         e_cols = MultiIndex.from_arrays(
             arrays=[[1, 1, 2, 2], [50, 100, 100, 200]], names=["col", "value"]
         )
-        expected = pd.DataFrame(data=e_data, index=e_index, columns=e_cols)
 
     else:
         e_data = [
@@ -290,6 +289,6 @@ def test_pivot_table_multiindex_values_as_two_params(index, columns, request):
             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)
 
+    expected = pd.DataFrame(data=e_data, index=e_index, columns=e_cols)
     tm.assert_frame_equal(left=result, right=expected)

Original file line number	Diff line number	Diff line change
`@@ -277,7 +277,6 @@ def test_pivot_table_multiindex_values_as_two_params(index, columns, request):`
`277`	`277`	`e_cols = MultiIndex.from_arrays(`
`278`	`278`	`arrays=[[1, 1, 2, 2], [50, 100, 100, 200]], names=["col", "value"]`
`279`	`279`	`)`
`280`		`- expected = pd.DataFrame(data=e_data, index=e_index, columns=e_cols)`
`281`	`280`
`282`	`281`	`else:`
`283`	`282`	`e_data = [`
`@@ -290,6 +289,6 @@ def test_pivot_table_multiindex_values_as_two_params(index, columns, request):`
`290`	`289`	`arrays=[["A", "A", "B", "B"], [50, 100, 100, 200]], names=["index", "value"]`
`291`	`290`	`)`
`292`	`291`	`e_cols = Index(data=[1, 2], name="col")`
`293`		`- expected = pd.DataFrame(data=e_data, index=e_index, columns=e_cols)`
`294`	`292`
	`293`	`+ expected = pd.DataFrame(data=e_data, index=e_index, columns=e_cols)`
`295`	`294`	`tm.assert_frame_equal(left=result, right=expected)`