@@ -2905,7 +2905,9 @@ def test_groupby_groups_datetimeindex_tz(self):
29052905 dates = ['2011-07-19 07:00:00' , '2011-07-19 08:00:00' , '2011-07-19 09:00:00' ,
29062906 '2011-07-19 07:00:00' , '2011-07-19 08:00:00' , '2011-07-19 09:00:00' ]
29072907 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 })
29092911 df ['datetime' ] = df ['datetime' ].apply (lambda d : Timestamp (d , tz = 'US/Pacific' ))
29102912
29112913 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):
29222924
29232925 # by level
29242926 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 )
29262930
29272931 exp_idx = pd .DatetimeIndex (['2011-07-19 07:00:00' , '2011-07-19 08:00:00' ,
29282932 '2011-07-19 09:00:00' ], tz = 'Asia/Tokyo' )
0 commit comments