@@ -180,19 +180,35 @@ def setup(self, method):
180
180
raise NotImplementedError
181
181
win = 100
182
182
arr = np .random .rand (100000 )
183
- rolling = {'rolling_median' : rolling_median ,
184
- 'rolling_mean' : rolling_mean ,
185
- 'rolling_min' : rolling_min ,
186
- 'rolling_max' : rolling_max ,
187
- 'rolling_var' : rolling_var ,
188
- 'rolling_skew' : rolling_skew ,
189
- 'rolling_kurt' : rolling_kurt ,
190
- 'rolling_std' : rolling_std }
191
-
192
- @test_parallel (num_threads = 2 )
193
- def parallel_rolling ():
194
- rolling [method ](arr , win )
195
- self .parallel_rolling = parallel_rolling
183
+ if hasattr (DataFrame , 'rolling' ):
184
+ rolling = {'rolling_median' : 'median' ,
185
+ 'rolling_mean' : 'mean' ,
186
+ 'rolling_min' : 'min' ,
187
+ 'rolling_max' : 'max' ,
188
+ 'rolling_var' : 'var' ,
189
+ 'rolling_skew' : 'skew' ,
190
+ 'rolling_kurt' : 'kurt' ,
191
+ 'rolling_std' : 'std' }
192
+ df = DataFrame (arr ).rolling (win )
193
+
194
+ @test_parallel (num_threads = 2 )
195
+ def parallel_rolling ():
196
+ getattr (df , rolling [method ])()
197
+ self .parallel_rolling = parallel_rolling
198
+ else :
199
+ rolling = {'rolling_median' : rolling_median ,
200
+ 'rolling_mean' : rolling_mean ,
201
+ 'rolling_min' : rolling_min ,
202
+ 'rolling_max' : rolling_max ,
203
+ 'rolling_var' : rolling_var ,
204
+ 'rolling_skew' : rolling_skew ,
205
+ 'rolling_kurt' : rolling_kurt ,
206
+ 'rolling_std' : rolling_std }
207
+
208
+ @test_parallel (num_threads = 2 )
209
+ def parallel_rolling ():
210
+ rolling [method ](arr , win )
211
+ self .parallel_rolling = parallel_rolling
196
212
197
213
def time_rolling (self , method ):
198
214
self .parallel_rolling ()
0 commit comments