Skip to content

Commit 8ea05b6

Browse files
ryankarlosproost
authored andcommitted
TST: Merge multiple cols with mixed columns/index (pandas-dev#29522)
1 parent 055a19d commit 8ea05b6

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

pandas/tests/reshape/merge/test_multi.py

+21
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,27 @@ def test_merge_right_vs_left(self, left, right, sort):
195195

196196
tm.assert_frame_equal(merged_left_right, merge_right_left)
197197

198+
def test_merge_multiple_cols_with_mixed_cols_index(self):
199+
# GH29522
200+
s = pd.Series(
201+
range(6),
202+
pd.MultiIndex.from_product([["A", "B"], [1, 2, 3]], names=["lev1", "lev2"]),
203+
name="Amount",
204+
)
205+
df = pd.DataFrame(
206+
{"lev1": list("AAABBB"), "lev2": [1, 2, 3, 1, 2, 3], "col": 0}
207+
)
208+
result = pd.merge(df, s.reset_index(), on=["lev1", "lev2"])
209+
expected = pd.DataFrame(
210+
{
211+
"lev1": list("AAABBB"),
212+
"lev2": [1, 2, 3, 1, 2, 3],
213+
"col": [0] * 6,
214+
"Amount": range(6),
215+
}
216+
)
217+
tm.assert_frame_equal(result, expected)
218+
198219
def test_compress_group_combinations(self):
199220

200221
# ~ 40000000 possible unique groups

0 commit comments

Comments
 (0)