Skip to content

Commit dbe3579

Browse files
Shivam Ranajreback
Shivam Rana
authored andcommitted
9236: test for the DataFrame.groupby with MultiIndex having pd.NaT (#25310)
1 parent 9561b96 commit dbe3579

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

pandas/tests/groupby/test_groupby.py

+16
Original file line numberDiff line numberDiff line change
@@ -1698,3 +1698,19 @@ def test_groupby_agg_ohlc_non_first():
16981698
result = df.groupby(pd.Grouper(freq='D')).agg(['sum', 'ohlc'])
16991699

17001700
tm.assert_frame_equal(result, expected)
1701+
1702+
1703+
def test_groupby_multiindex_nat():
1704+
# GH 9236
1705+
values = [
1706+
(pd.NaT, 'a'),
1707+
(datetime(2012, 1, 2), 'a'),
1708+
(datetime(2012, 1, 2), 'b'),
1709+
(datetime(2012, 1, 3), 'a')
1710+
]
1711+
mi = pd.MultiIndex.from_tuples(values, names=['date', None])
1712+
ser = pd.Series([3, 2, 2.5, 4], index=mi)
1713+
1714+
result = ser.groupby(level=1).mean()
1715+
expected = pd.Series([3., 2.5], index=["a", "b"])
1716+
assert_series_equal(result, expected)

0 commit comments

Comments
 (0)