Skip to content

Commit b28f9b1

Browse files
committed
Remove duplicate levels from test_frame_describe_multikey
1 parent 1c33751 commit b28f9b1

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

pandas/tests/groupby/test_groupby.py

+5-6
Original file line numberDiff line numberDiff line change
@@ -756,10 +756,9 @@ def test_frame_describe_multikey(self):
756756
desc_groups = []
757757
for col in self.tsframe:
758758
group = grouped[col].describe()
759-
group_col = pd.MultiIndex([[col] * len(group.columns),
760-
group.columns],
761-
[[0] * len(group.columns),
762-
range(len(group.columns))])
759+
group_col = pd.MultiIndex(
760+
levels=[[col], group.columns],
761+
labels=[[0]*len(group.columns), range(len(group.columns))])
763762
group = pd.DataFrame(group.values,
764763
columns=group_col,
765764
index=group.index)
@@ -771,8 +770,8 @@ def test_frame_describe_multikey(self):
771770
'C': 1, 'D': 1}, axis=1)
772771
result = groupedT.describe()
773772
expected = self.tsframe.describe().T
774-
expected.index = pd.MultiIndex([[0, 0, 1, 1], expected.index],
775-
[range(4), range(len(expected.index))])
773+
expected.index = pd.MultiIndex(levels=[[0, 1], expected.index],
774+
labels=[[0, 0, 1, 1], range(len(expected.index))])
776775
tm.assert_frame_equal(result, expected)
777776

778777
def test_frame_describe_tupleindex(self):

0 commit comments

Comments
 (0)