Skip to content

Commit 720e309

Browse files
committed
add test for MI
1 parent df0fccf commit 720e309

File tree

2 files changed

+31
-1
lines changed

2 files changed

+31
-1
lines changed

pandas/tests/frame/test_explode.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,36 @@ def test_basic():
3333
tm.assert_frame_equal(result, expected)
3434

3535

36+
def test_multi_index():
37+
df = pd.DataFrame(
38+
{"A": np.array([[0, 1, 2], np.nan, [], (3, 4)], dtype=object), "B": 1},
39+
index=pd.MultiIndex.from_tuples([("a", 1), ("a", 2), ("b", 1), ("b", 2)]),
40+
)
41+
42+
result = df.explode("A")
43+
expected = pd.DataFrame(
44+
{
45+
"A": pd.Series(
46+
[0, 1, 2, np.nan, np.nan, 3, 4],
47+
index=pd.MultiIndex.from_tuples(
48+
[
49+
("a", 1),
50+
("a", 1),
51+
("a", 1),
52+
("a", 2),
53+
("b", 1),
54+
("b", 2),
55+
("b", 2),
56+
]
57+
),
58+
dtype=object,
59+
),
60+
"B": 1,
61+
}
62+
)
63+
tm.assert_frame_equal(result, expected)
64+
65+
3666
def test_usecase():
3767
# explode a single column
3868
# gh-10511

pandas/tests/series/test_explode.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def test_typical_usecase():
9090
expected = pd.DataFrame(
9191
{"var2": [1, 1, 1, 2, 2, 2], "var1": list("abcdef")}, index=[0, 0, 0, 1, 1, 1]
9292
)
93-
tm.assert_frame_equal(result, expected, check_like=True)
93+
tm.assert_frame_equal(result, expected)
9494

9595

9696
def test_nested_EA():

0 commit comments

Comments
 (0)