@@ -444,13 +444,22 @@ def test_groupby_grouper_f_sanity_checked(self):
444
444
445
445
def test_groupby_with_datetime_key (self ):
446
446
# GH 51158
447
- df = DataFrame ({"id" : ["a" , "b" ] * 3 , "b" : pd .date_range ("2000-01-01" , "2000-01-03" , freq = "9H" )})
448
- grouper = pd .Grouper (key = 'b' , freq = "D" )
449
- g = df .groupby ([grouper , 'id' ])
447
+ df = DataFrame (
448
+ {
449
+ "id" : ["a" , "b" ] * 3 ,
450
+ "b" : date_range ("2000-01-01" , "2000-01-03" , freq = "9H" ),
451
+ }
452
+ )
453
+ grouper = Grouper (key = "b" , freq = "D" )
454
+ g = df .groupby ([grouper , "id" ])
450
455
451
456
# test number of groups
452
- expected = {(pd .Timestamp ("2000-01-01" ), 'a' ): [0 , 2 ], (pd .Timestamp ("2000-01-01" ), 'b' ): [1 ],
453
- (pd .Timestamp ("2000-01-02" ), 'a' ): [4 ], (pd .Timestamp ("2000-01-02" ), 'b' ): [3 , 5 ]}
457
+ expected = {
458
+ (Timestamp ("2000-01-01" ), "a" ): [0 , 2 ],
459
+ (Timestamp ("2000-01-01" ), "b" ): [1 ],
460
+ (Timestamp ("2000-01-02" ), "a" ): [4 ],
461
+ (Timestamp ("2000-01-02" ), "b" ): [3 , 5 ],
462
+ }
454
463
tm .assert_dict_equal (g .groups , expected )
455
464
456
465
# test number of group keys
0 commit comments