Skip to content

Commit 87c1761

Browse files
authored
adding test for #18451 (#35494)
1 parent 2a31923 commit 87c1761

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

pandas/tests/groupby/test_groupby.py

+14
Original file line numberDiff line numberDiff line change
@@ -2055,3 +2055,17 @@ def test_groups_repr_truncates(max_seq_items, expected):
20552055

20562056
result = df.groupby(np.array(df.a)).groups.__repr__()
20572057
assert result == expected
2058+
2059+
2060+
def test_group_on_two_row_multiindex_returns_one_tuple_key():
2061+
# GH 18451
2062+
df = pd.DataFrame([{"a": 1, "b": 2, "c": 99}, {"a": 1, "b": 2, "c": 88}])
2063+
df = df.set_index(["a", "b"])
2064+
2065+
grp = df.groupby(["a", "b"])
2066+
result = grp.indices
2067+
expected = {(1, 2): np.array([0, 1], dtype=np.int64)}
2068+
2069+
assert len(result) == 1
2070+
key = (1, 2)
2071+
assert (result[key] == expected[key]).all()

0 commit comments

Comments
 (0)