Skip to content

Commit c169645

Browse files
committed
Updated test_frame_describe_multikey to remove duplicate MultiIndex levels
1 parent 386daaf commit c169645

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

pandas/tests/groupby/test_functional.py

+9-6
Original file line numberDiff line numberDiff line change
@@ -46,16 +46,17 @@ def test_series_index_name(self):
4646
result = grouped.agg(lambda x: x.mean())
4747
assert result.index.name == 'A'
4848

49+
4950
def test_frame_describe_multikey(self):
5051
grouped = self.tsframe.groupby([lambda x: x.year, lambda x: x.month])
5152
result = grouped.describe()
5253
desc_groups = []
5354
for col in self.tsframe:
5455
group = grouped[col].describe()
55-
group_col = pd.MultiIndex([[col] * len(group.columns),
56-
group.columns],
57-
[[0] * len(group.columns),
58-
range(len(group.columns))])
56+
# GH 17464 - Remove duplicate MultiIndex levels
57+
group_col = pd.MultiIndex(
58+
levels=[[col], group.columns],
59+
labels=[[0] * len(group.columns), range(len(group.columns))])
5960
group = pd.DataFrame(group.values,
6061
columns=group_col,
6162
index=group.index)
@@ -67,8 +68,10 @@ def test_frame_describe_multikey(self):
6768
'C': 1, 'D': 1}, axis=1)
6869
result = groupedT.describe()
6970
expected = self.tsframe.describe().T
70-
expected.index = pd.MultiIndex([[0, 0, 1, 1], expected.index],
71-
[range(4), range(len(expected.index))])
71+
# GH 17464 - Remove duplicate MultiIndex levels
72+
expected.index = pd.MultiIndex(
73+
levels=[[0, 1], expected.index],
74+
labels=[[0, 0, 1, 1], range(len(expected.index))])
7275
tm.assert_frame_equal(result, expected)
7376

7477
def test_frame_describe_tupleindex(self):

0 commit comments

Comments
 (0)