From c1696459c9fe3ad3dca4aa3e896cc425069dfd18 Mon Sep 17 00:00:00 2001 From: Chris Mazzullo Date: Tue, 28 Nov 2017 22:54:52 -0500 Subject: [PATCH] Updated `test_frame_describe_multikey` to remove duplicate MultiIndex levels --- pandas/tests/groupby/test_functional.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/pandas/tests/groupby/test_functional.py b/pandas/tests/groupby/test_functional.py index bc13d51c4f4f6..dedc072a8d4ae 100644 --- a/pandas/tests/groupby/test_functional.py +++ b/pandas/tests/groupby/test_functional.py @@ -46,16 +46,17 @@ def test_series_index_name(self): result = grouped.agg(lambda x: x.mean()) assert result.index.name == 'A' + def test_frame_describe_multikey(self): grouped = self.tsframe.groupby([lambda x: x.year, lambda x: x.month]) result = grouped.describe() desc_groups = [] for col in self.tsframe: group = grouped[col].describe() - group_col = pd.MultiIndex([[col] * len(group.columns), - group.columns], - [[0] * len(group.columns), - range(len(group.columns))]) + # GH 17464 - Remove duplicate MultiIndex levels + group_col = pd.MultiIndex( + levels=[[col], group.columns], + labels=[[0] * len(group.columns), range(len(group.columns))]) group = pd.DataFrame(group.values, columns=group_col, index=group.index) @@ -67,8 +68,10 @@ def test_frame_describe_multikey(self): 'C': 1, 'D': 1}, axis=1) result = groupedT.describe() expected = self.tsframe.describe().T - expected.index = pd.MultiIndex([[0, 0, 1, 1], expected.index], - [range(4), range(len(expected.index))]) + # GH 17464 - Remove duplicate MultiIndex levels + expected.index = pd.MultiIndex( + levels=[[0, 1], expected.index], + labels=[[0, 0, 1, 1], range(len(expected.index))]) tm.assert_frame_equal(result, expected) def test_frame_describe_tupleindex(self):