1
1
from .pandas_vb_common import *
2
- from pandas .core import common as com
2
+
3
+ from pandas .core .algorithms import take_1d
3
4
4
5
try :
5
6
from cStringIO import StringIO
6
7
except ImportError :
7
8
from io import StringIO
8
9
10
+ try :
11
+ from pandas ._libs import algos
12
+ except ImportError :
13
+ from pandas import algos
14
+
9
15
try :
10
16
from pandas .util .testing import test_parallel
11
17
@@ -167,11 +173,11 @@ def time_nogil_take1d_float64(self):
167
173
168
174
@test_parallel (num_threads = 2 )
169
175
def take_1d_pg2_int64 (self ):
170
- com . take_1d (self .df .int64 .values , self .indexer )
176
+ take_1d (self .df .int64 .values , self .indexer )
171
177
172
178
@test_parallel (num_threads = 2 )
173
179
def take_1d_pg2_float64 (self ):
174
- com . take_1d (self .df .float64 .values , self .indexer )
180
+ take_1d (self .df .float64 .values , self .indexer )
175
181
176
182
177
183
class nogil_take1d_int64 (object ):
@@ -193,11 +199,11 @@ def time_nogil_take1d_int64(self):
193
199
194
200
@test_parallel (num_threads = 2 )
195
201
def take_1d_pg2_int64 (self ):
196
- com . take_1d (self .df .int64 .values , self .indexer )
202
+ take_1d (self .df .int64 .values , self .indexer )
197
203
198
204
@test_parallel (num_threads = 2 )
199
205
def take_1d_pg2_float64 (self ):
200
- com . take_1d (self .df .float64 .values , self .indexer )
206
+ take_1d (self .df .float64 .values , self .indexer )
201
207
202
208
203
209
class nogil_kth_smallest (object ):
@@ -226,7 +232,7 @@ class nogil_datetime_fields(object):
226
232
227
233
def setup (self ):
228
234
self .N = 100000000
229
- self .dti = pd .date_range ('1900-01-01' , periods = self .N , freq = 'D ' )
235
+ self .dti = pd .date_range ('1900-01-01' , periods = self .N , freq = 'T ' )
230
236
self .period = self .dti .to_period ('D' )
231
237
if (not have_real_test_parallel ):
232
238
raise NotImplementedError
0 commit comments