36
36
from .pandas_vb_common import BaseIO # isort:skip
37
37
38
38
39
- def test_parallel (num_threads = 2 , kwargs_list = None ):
39
+ def run_parallel (num_threads = 2 , kwargs_list = None ):
40
40
"""
41
41
Decorator to run the same function multiple times in parallel.
42
42
@@ -95,7 +95,7 @@ def setup(self, threads, method):
95
95
{"key" : np .random .randint (0 , ngroups , size = N ), "data" : np .random .randn (N )}
96
96
)
97
97
98
- @test_parallel (num_threads = threads )
98
+ @run_parallel (num_threads = threads )
99
99
def parallel ():
100
100
getattr (df .groupby ("key" )["data" ], method )()
101
101
@@ -123,7 +123,7 @@ def setup(self, threads):
123
123
ngroups = 10 ** 3
124
124
data = Series (np .random .randint (0 , ngroups , size = size ))
125
125
126
- @test_parallel (num_threads = threads )
126
+ @run_parallel (num_threads = threads )
127
127
def get_groups ():
128
128
data .groupby (data ).groups
129
129
@@ -142,7 +142,7 @@ def setup(self, dtype):
142
142
df = DataFrame ({"col" : np .arange (N , dtype = dtype )})
143
143
indexer = np .arange (100 , len (df ) - 100 )
144
144
145
- @test_parallel (num_threads = 2 )
145
+ @run_parallel (num_threads = 2 )
146
146
def parallel_take1d ():
147
147
take_nd (df ["col" ].values , indexer )
148
148
@@ -163,7 +163,7 @@ def setup(self):
163
163
k = 5 * 10 ** 5
164
164
kwargs_list = [{"arr" : np .random .randn (N )}, {"arr" : np .random .randn (N )}]
165
165
166
- @test_parallel (num_threads = 2 , kwargs_list = kwargs_list )
166
+ @run_parallel (num_threads = 2 , kwargs_list = kwargs_list )
167
167
def parallel_kth_smallest (arr ):
168
168
algos .kth_smallest (arr , k )
169
169
@@ -180,42 +180,42 @@ def setup(self):
180
180
self .period = self .dti .to_period ("D" )
181
181
182
182
def time_datetime_field_year (self ):
183
- @test_parallel (num_threads = 2 )
183
+ @run_parallel (num_threads = 2 )
184
184
def run (dti ):
185
185
dti .year
186
186
187
187
run (self .dti )
188
188
189
189
def time_datetime_field_day (self ):
190
- @test_parallel (num_threads = 2 )
190
+ @run_parallel (num_threads = 2 )
191
191
def run (dti ):
192
192
dti .day
193
193
194
194
run (self .dti )
195
195
196
196
def time_datetime_field_daysinmonth (self ):
197
- @test_parallel (num_threads = 2 )
197
+ @run_parallel (num_threads = 2 )
198
198
def run (dti ):
199
199
dti .days_in_month
200
200
201
201
run (self .dti )
202
202
203
203
def time_datetime_field_normalize (self ):
204
- @test_parallel (num_threads = 2 )
204
+ @run_parallel (num_threads = 2 )
205
205
def run (dti ):
206
206
dti .normalize ()
207
207
208
208
run (self .dti )
209
209
210
210
def time_datetime_to_period (self ):
211
- @test_parallel (num_threads = 2 )
211
+ @run_parallel (num_threads = 2 )
212
212
def run (dti ):
213
213
dti .to_period ("s" )
214
214
215
215
run (self .dti )
216
216
217
217
def time_period_to_datetime (self ):
218
- @test_parallel (num_threads = 2 )
218
+ @run_parallel (num_threads = 2 )
219
219
def run (period ):
220
220
period .to_timestamp ()
221
221
@@ -232,7 +232,7 @@ def setup(self, method):
232
232
if hasattr (DataFrame , "rolling" ):
233
233
df = DataFrame (arr ).rolling (win )
234
234
235
- @test_parallel (num_threads = 2 )
235
+ @run_parallel (num_threads = 2 )
236
236
def parallel_rolling ():
237
237
getattr (df , method )()
238
238
@@ -249,7 +249,7 @@ def parallel_rolling():
249
249
"std" : rolling_std ,
250
250
}
251
251
252
- @test_parallel (num_threads = 2 )
252
+ @run_parallel (num_threads = 2 )
253
253
def parallel_rolling ():
254
254
rolling [method ](arr , win )
255
255
@@ -286,7 +286,7 @@ def setup(self, dtype):
286
286
self .fname = f"__test_{ dtype } __.csv"
287
287
df .to_csv (self .fname )
288
288
289
- @test_parallel (num_threads = 2 )
289
+ @run_parallel (num_threads = 2 )
290
290
def parallel_read_csv ():
291
291
read_csv (self .fname )
292
292
@@ -305,7 +305,7 @@ class ParallelFactorize:
305
305
def setup (self , threads ):
306
306
strings = Index ([f"i-{ i } " for i in range (100000 )], dtype = object )
307
307
308
- @test_parallel (num_threads = threads )
308
+ @run_parallel (num_threads = threads )
309
309
def parallel ():
310
310
factorize (strings )
311
311
0 commit comments