File tree 1 file changed +12
-1
lines changed
1 file changed +12
-1
lines changed Original file line number Diff line number Diff line change @@ -27,16 +27,27 @@ def time_constructor(self, data):
27
27
class IsIn (object ):
28
28
29
29
goal_time = 0.2
30
- params = ['int64' , 'object' ]
30
+ params = ['int64' , 'object' , 'float64' ]
31
31
param_names = ['dtype' ]
32
32
33
33
def setup (self , dtype ):
34
34
self .s = Series (np .random .randint (1 , 10 , 100000 )).astype (dtype )
35
35
self .values = [1 , 2 ]
36
+ self .small = Series (np .random .randint (1 , 10 , 10 )).astype (dtype )
37
+ self .many_different_values = np .arange (10 ** 6 ).astype (dtype )
38
+ self .few_different_values = np .zeros (10 ** 6 ).astype (dtype )
36
39
37
40
def time_isin (self , dtypes ):
38
41
self .s .isin (self .values )
39
42
43
+ def time_isin_many_different (self , dtypes ):
44
+ # runtime is dominated by creation of the lookup-table
45
+ self .small .isin (self .many_different_values )
46
+
47
+ def time_isin_few_different (self , dtypes ):
48
+ # runtime is dominated by creation of the lookup-table
49
+ self .small .isin (self .few_different_values )
50
+
40
51
41
52
class NSort (object ):
42
53
You can’t perform that action at this time.
0 commit comments