11import numpy as np
22
3- try :
4- from pandas .compat import np_version_under1p20
5- except ImportError :
6- from pandas .compat .numpy import _np_version_under1p20 as np_version_under1p20
7-
83from pandas import (
94 Categorical ,
105 NaT ,
@@ -283,10 +278,6 @@ class IsInLongSeriesLookUpDominates:
283278 def setup (self , dtype , MaxNumber , series_type ):
284279 N = 10 ** 7
285280
286- # https://github.com/pandas-dev/pandas/issues/39844
287- if not np_version_under1p20 and dtype in ("Int64" , "Float64" ):
288- raise NotImplementedError
289-
290281 if series_type == "random_hits" :
291282 array = np .random .randint (0 , MaxNumber , N )
292283 if series_type == "random_misses" :
@@ -297,7 +288,8 @@ def setup(self, dtype, MaxNumber, series_type):
297288 array = np .arange (N ) + MaxNumber
298289
299290 self .series = Series (array ).astype (dtype )
300- self .values = np .arange (MaxNumber ).astype (dtype )
291+
292+ self .values = np .arange (MaxNumber ).astype (dtype .lower ())
301293
302294 def time_isin (self , dtypes , MaxNumber , series_type ):
303295 self .series .isin (self .values )
@@ -313,16 +305,12 @@ class IsInLongSeriesValuesDominate:
313305 def setup (self , dtype , series_type ):
314306 N = 10 ** 7
315307
316- # https://github.com/pandas-dev/pandas/issues/39844
317- if not np_version_under1p20 and dtype in ("Int64" , "Float64" ):
318- raise NotImplementedError
319-
320308 if series_type == "random" :
321309 vals = np .random .randint (0 , 10 * N , N )
322310 if series_type == "monotone" :
323311 vals = np .arange (N )
324312
325- self .values = vals .astype (dtype )
313+ self .values = vals .astype (dtype . lower () )
326314 M = 10 ** 6 + 1
327315 self .series = Series (np .arange (M )).astype (dtype )
328316
0 commit comments