@@ -2905,7 +2905,9 @@ def test_groupby_groups_datetimeindex_tz(self):
2905
2905
dates = ['2011-07-19 07:00:00' , '2011-07-19 08:00:00' , '2011-07-19 09:00:00' ,
2906
2906
'2011-07-19 07:00:00' , '2011-07-19 08:00:00' , '2011-07-19 09:00:00' ]
2907
2907
df = DataFrame ({'label' : ['a' , 'a' , 'a' , 'b' , 'b' , 'b' ],
2908
- 'datetime' : dates , 'value1' : range (6 ), 'value2' : [1 , 2 ] * 3 })
2908
+ 'datetime' : dates ,
2909
+ 'value1' : np .arange (6 ,dtype = 'int64' ),
2910
+ 'value2' : [1 , 2 ] * 3 })
2909
2911
df ['datetime' ] = df ['datetime' ].apply (lambda d : Timestamp (d , tz = 'US/Pacific' ))
2910
2912
2911
2913
exp_idx1 = pd .DatetimeIndex (['2011-07-19 07:00:00' , '2011-07-19 07:00:00' ,
@@ -2922,7 +2924,9 @@ def test_groupby_groups_datetimeindex_tz(self):
2922
2924
2923
2925
# by level
2924
2926
didx = pd .DatetimeIndex (dates , tz = 'Asia/Tokyo' )
2925
- df = DataFrame ({'value1' : range (6 ), 'value2' : [1 , 2 , 3 , 1 , 2 , 3 ]}, index = didx )
2927
+ df = DataFrame ({'value1' : np .arange (6 ,dtype = 'int64' ),
2928
+ 'value2' : [1 , 2 , 3 , 1 , 2 , 3 ]},
2929
+ index = didx )
2926
2930
2927
2931
exp_idx = pd .DatetimeIndex (['2011-07-19 07:00:00' , '2011-07-19 08:00:00' ,
2928
2932
'2011-07-19 09:00:00' ], tz = 'Asia/Tokyo' )
0 commit comments