File tree 1 file changed +20
-0
lines changed
1 file changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -543,6 +543,26 @@ def predictions(tool):
543
543
tm .assert_series_equal (expected , result )
544
544
545
545
546
+ def test_apply_aggregating_timedelta_and_datetime ():
547
+ # Regression test for GH 15562
548
+ # The following groupby caused ValueErrors and IndexErrors pre 0.20.0
549
+
550
+ s = """clientid,datetime
551
+ A,2017-02-01 00:00:00
552
+ B,2017-02-01 00:00:00
553
+ C,2017-02-01 00:00:00"""
554
+
555
+ df = pd .read_csv (StringIO (s ))
556
+ df ["datetime" ] = pd .to_datetime (df .datetime )
557
+ df ["time_delta_zero" ] = df .datetime - df .datetime
558
+ result = df .groupby ("clientid" ).apply (
559
+ lambda ddf : pd .Series (
560
+ dict (clientid_age = ddf .time_delta_zero .min (), date = ddf .datetime .min ())
561
+ )
562
+ )
563
+ assert result .shape == (3 , 2 )
564
+
565
+
546
566
def test_time_field_bug ():
547
567
# Test a fix for the following error related to GH issue 11324 When
548
568
# non-key fields in a group-by dataframe contained time-based fields
You can’t perform that action at this time.
0 commit comments