Skip to content

Commit 5e1a79b

Browse files
committed
Add current rolling to gil.py
1 parent bbb33bc commit 5e1a79b

File tree

1 file changed

+29
-13
lines changed

1 file changed

+29
-13
lines changed

asv_bench/benchmarks/gil.py

+29-13
Original file line numberDiff line numberDiff line change
@@ -180,19 +180,35 @@ def setup(self, method):
180180
raise NotImplementedError
181181
win = 100
182182
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
196212

197213
def time_rolling(self, method):
198214
self.parallel_rolling()

0 commit comments

Comments
 (0)