3636from .pandas_vb_common import BaseIO # isort:skip
3737
3838
39- def test_parallel (num_threads = 2 , kwargs_list = None ):
39+ def run_parallel (num_threads = 2 , kwargs_list = None ):
4040 """
4141 Decorator to run the same function multiple times in parallel.
4242
@@ -95,7 +95,7 @@ def setup(self, threads, method):
9595 {"key" : np .random .randint (0 , ngroups , size = N ), "data" : np .random .randn (N )}
9696 )
9797
98- @test_parallel (num_threads = threads )
98+ @run_parallel (num_threads = threads )
9999 def parallel ():
100100 getattr (df .groupby ("key" )["data" ], method )()
101101
@@ -123,7 +123,7 @@ def setup(self, threads):
123123 ngroups = 10 ** 3
124124 data = Series (np .random .randint (0 , ngroups , size = size ))
125125
126- @test_parallel (num_threads = threads )
126+ @run_parallel (num_threads = threads )
127127 def get_groups ():
128128 data .groupby (data ).groups
129129
@@ -142,7 +142,7 @@ def setup(self, dtype):
142142 df = DataFrame ({"col" : np .arange (N , dtype = dtype )})
143143 indexer = np .arange (100 , len (df ) - 100 )
144144
145- @test_parallel (num_threads = 2 )
145+ @run_parallel (num_threads = 2 )
146146 def parallel_take1d ():
147147 take_nd (df ["col" ].values , indexer )
148148
@@ -163,7 +163,7 @@ def setup(self):
163163 k = 5 * 10 ** 5
164164 kwargs_list = [{"arr" : np .random .randn (N )}, {"arr" : np .random .randn (N )}]
165165
166- @test_parallel (num_threads = 2 , kwargs_list = kwargs_list )
166+ @run_parallel (num_threads = 2 , kwargs_list = kwargs_list )
167167 def parallel_kth_smallest (arr ):
168168 algos .kth_smallest (arr , k )
169169
@@ -180,42 +180,42 @@ def setup(self):
180180 self .period = self .dti .to_period ("D" )
181181
182182 def time_datetime_field_year (self ):
183- @test_parallel (num_threads = 2 )
183+ @run_parallel (num_threads = 2 )
184184 def run (dti ):
185185 dti .year
186186
187187 run (self .dti )
188188
189189 def time_datetime_field_day (self ):
190- @test_parallel (num_threads = 2 )
190+ @run_parallel (num_threads = 2 )
191191 def run (dti ):
192192 dti .day
193193
194194 run (self .dti )
195195
196196 def time_datetime_field_daysinmonth (self ):
197- @test_parallel (num_threads = 2 )
197+ @run_parallel (num_threads = 2 )
198198 def run (dti ):
199199 dti .days_in_month
200200
201201 run (self .dti )
202202
203203 def time_datetime_field_normalize (self ):
204- @test_parallel (num_threads = 2 )
204+ @run_parallel (num_threads = 2 )
205205 def run (dti ):
206206 dti .normalize ()
207207
208208 run (self .dti )
209209
210210 def time_datetime_to_period (self ):
211- @test_parallel (num_threads = 2 )
211+ @run_parallel (num_threads = 2 )
212212 def run (dti ):
213213 dti .to_period ("s" )
214214
215215 run (self .dti )
216216
217217 def time_period_to_datetime (self ):
218- @test_parallel (num_threads = 2 )
218+ @run_parallel (num_threads = 2 )
219219 def run (period ):
220220 period .to_timestamp ()
221221
@@ -232,7 +232,7 @@ def setup(self, method):
232232 if hasattr (DataFrame , "rolling" ):
233233 df = DataFrame (arr ).rolling (win )
234234
235- @test_parallel (num_threads = 2 )
235+ @run_parallel (num_threads = 2 )
236236 def parallel_rolling ():
237237 getattr (df , method )()
238238
@@ -249,7 +249,7 @@ def parallel_rolling():
249249 "std" : rolling_std ,
250250 }
251251
252- @test_parallel (num_threads = 2 )
252+ @run_parallel (num_threads = 2 )
253253 def parallel_rolling ():
254254 rolling [method ](arr , win )
255255
@@ -286,7 +286,7 @@ def setup(self, dtype):
286286 self .fname = f"__test_{ dtype } __.csv"
287287 df .to_csv (self .fname )
288288
289- @test_parallel (num_threads = 2 )
289+ @run_parallel (num_threads = 2 )
290290 def parallel_read_csv ():
291291 read_csv (self .fname )
292292
@@ -305,7 +305,7 @@ class ParallelFactorize:
305305 def setup (self , threads ):
306306 strings = Index ([f"i-{ i } " for i in range (100000 )], dtype = object )
307307
308- @test_parallel (num_threads = threads )
308+ @run_parallel (num_threads = threads )
309309 def parallel ():
310310 factorize (strings )
311311
0 commit comments