Skip to content

Commit 1aebb81

Browse files
committed
added test for issue GH 8866
1 parent 6a94388 commit 1aebb81

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

pandas/core/groupby.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,6 @@ def _set_grouper(self, obj, sort=False):
245245
return self.grouper
246246
else:
247247
self._set_basegrouper(obj)
248-
return self.grouper
249248

250249
def _set_timegrouper(self, obj, sort=False):
251250
"""
@@ -350,6 +349,7 @@ def is_in_obj(gpr):
350349
grouper = BaseGrouper(ax, grouping)
351350
self.obj = obj
352351
self.grouper = grouper
352+
return self.grouper
353353

354354
def _get_binner_for_grouping(self, obj):
355355
raise NotImplementedError

pandas/tests/test_groupby.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3677,6 +3677,16 @@ def test_timegrouper_get_group(self):
36773677
dt = pd.Timestamp(t)
36783678
result = grouped.get_group(dt)
36793679
assert_frame_equal(result, expected)
3680+
3681+
def test_grouper_with_nondatetime(self):
3682+
# GH 8866
3683+
s = Series(np.arange(8),index=pd.MultiIndex.from_product([list('ab'),
3684+
range(2),pd.date_range('20130101',periods=2)],
3685+
names=['one','two','three']))
3686+
3687+
expected = Series(data = [6,22], index=pd.Index(['a','b'], name='one'))
3688+
result = s.groupby(pd.Grouper(level='one')).sum()
3689+
assert_series_equal(result,expected)
36803690

36813691
def test_cumcount(self):
36823692
df = DataFrame([['a'], ['a'], ['a'], ['b'], ['a']], columns=['A'])

0 commit comments

Comments
 (0)