TST: windows dtype fix related GH3950

pandas-dev · May 13, 2014 · 4c1f036 · 4c1f036
1 parent 2bb2a8d
commit 4c1f036
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pandas/tests/test_groupby.py b/pandas/tests/test_groupby.py
@@ -2905,7 +2905,9 @@ def test_groupby_groups_datetimeindex_tz(self):
         dates = ['2011-07-19 07:00:00', '2011-07-19 08:00:00', '2011-07-19 09:00:00',
                  '2011-07-19 07:00:00', '2011-07-19 08:00:00', '2011-07-19 09:00:00']
         df = DataFrame({'label': ['a', 'a', 'a', 'b', 'b', 'b'],
-                         'datetime': dates, 'value1': range(6), 'value2': [1, 2] * 3})
+                        'datetime': dates, 
+                        'value1': np.arange(6,dtype='int64'), 
+                        'value2': [1, 2] * 3})
         df['datetime'] = df['datetime'].apply(lambda d: Timestamp(d, tz='US/Pacific'))
 
         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):
 
         # by level
         didx = pd.DatetimeIndex(dates, tz='Asia/Tokyo')
-        df = DataFrame({'value1': range(6), 'value2': [1, 2, 3, 1, 2, 3]}, index=didx)
+        df = DataFrame({'value1': np.arange(6,dtype='int64'), 
+                        'value2': [1, 2, 3, 1, 2, 3]}, 
+                       index=didx)
 
         exp_idx = pd.DatetimeIndex(['2011-07-19 07:00:00', '2011-07-19 08:00:00',
                                     '2011-07-19 09:00:00'], tz='Asia/Tokyo')