add test

Author: arw2019

pandas/tests/reshape/test_pivot_multilevel.py

@@ -2,7 +2,7 @@
 import pytest
 
 import pandas as pd
-from pandas import Index, MultiIndex
+from pandas import Index, Int64Index, MultiIndex
 import pandas._testing as tm
 
 
@@ -190,3 +190,35 @@ def test_pivot_list_like_columns(
         expected_values, columns=expected_columns, index=expected_index
     )
     tm.assert_frame_equal(result, expected)
+
+
+def test_pivot_multiindexed_rows_and_cols():
+    # GH 36360
+
+    df = pd.DataFrame(
+        data=np.arange(12).reshape(4, 3),
+        columns=MultiIndex.from_tuples(
+            [(0, 0), (0, 1), (0, 2)], names=["col_L0", "col_L1"]
+        ),
+        index=MultiIndex.from_tuples(
+            [(0, 0, 0), (0, 0, 1), (1, 1, 1), (1, 0, 0)],
+            names=["idx_L0", "idx_L1", "idx_L2"],
+        ),
+    )
+
+    res = df.pivot_table(
+        index=["idx_L0"],
+        columns=["idx_L1"],
+        values=[(0, 1)],
+        aggfunc=lambda col: col.values.sum(),
+    )
+
+    expected = pd.DataFrame(
+        data=[[5.0, np.nan], [10.0, 7.0]],
+        columns=MultiIndex.from_tuples(
+            [(0, 1, 0), (0, 1, 1)], names=["col_L0", "col_L1", "idx_L1"]
+        ),
+        index=Int64Index([0, 1], dtype="int64", name="idx_L0"),
+    )
+
+    tm.assert_frame_equal(res, expected)