Skip to content

Commit 05e6557

Browse files
author
Jon M. Mease
committed
Added test cases that group by column then index
1 parent 897ec1c commit 05e6557

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

pandas/tests/test_groupby.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,10 +474,21 @@ def test_grouper_column_and_index(self):
474474
expected = df_multi.reset_index().groupby(['B', 'inner']).mean()
475475
assert_frame_equal(result, expected)
476476

477+
# Test the reverse grouping order
478+
result = df_multi.groupby([pd.Grouper(level='inner'), 'B']).mean()
479+
expected = df_multi.reset_index().groupby(['inner', 'B']).mean()
480+
assert_frame_equal(result, expected)
481+
477482
# Grouping a single-index frame by a column and the index should
478483
# be equivalent to resetting the index and grouping by two columns
479484
df_single = df_multi.reset_index('outer')
480485
result = df_single.groupby(['B', pd.Grouper(level='inner')]).mean()
486+
expected = df_single.reset_index().groupby(['B', 'inner']).mean()
487+
assert_frame_equal(result, expected)
488+
489+
# Test the reverse grouping order
490+
result = df_single.groupby([pd.Grouper(level='inner'), 'B']).mean()
491+
expected = df_single.reset_index().groupby(['inner', 'B']).mean()
481492
assert_frame_equal(result, expected)
482493

483494
def test_grouper_getting_correct_binner(self):

0 commit comments

Comments
 (0)