Skip to content

Commit 3ece807

Browse files
authored
TST: groupby cumsum with named aggregate (#50033)
1 parent e61e58e commit 3ece807

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

pandas/tests/extension/base/groupby.py

+24
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,30 @@ def test_groupby_agg_extension(self, data_for_grouping):
5656
result = df.groupby("A").first()
5757
self.assert_frame_equal(result, expected)
5858

59+
def test_groupby_agg_extension_timedelta_cumsum_with_named_aggregation(self):
60+
# GH#41720
61+
expected = pd.DataFrame(
62+
{
63+
"td": {
64+
0: pd.Timedelta("0 days 01:00:00"),
65+
1: pd.Timedelta("0 days 01:15:00"),
66+
2: pd.Timedelta("0 days 01:15:00"),
67+
}
68+
}
69+
)
70+
df = pd.DataFrame(
71+
{
72+
"td": pd.Series(
73+
["0 days 01:00:00", "0 days 00:15:00", "0 days 01:15:00"],
74+
dtype="timedelta64[ns]",
75+
),
76+
"grps": ["a", "a", "b"],
77+
}
78+
)
79+
gb = df.groupby("grps")
80+
result = gb.agg(td=("td", "cumsum"))
81+
self.assert_frame_equal(result, expected)
82+
5983
def test_groupby_extension_no_sort(self, data_for_grouping):
6084
df = pd.DataFrame({"A": [1, 1, 2, 2, 3, 3, 1, 4], "B": data_for_grouping})
6185
result = df.groupby("B", sort=False).A.mean()

0 commit comments

Comments
 (0)