Skip to content

Commit d31bb00

Browse files
committed
adding performance tests for isin in combination with float64
1 parent c393015 commit d31bb00

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

asv_bench/benchmarks/series_methods.py

+12-1
Original file line numberDiff line numberDiff line change
@@ -27,16 +27,27 @@ def time_constructor(self, data):
2727
class IsIn(object):
2828

2929
goal_time = 0.2
30-
params = ['int64', 'object']
30+
params = ['int64', 'object', 'float64']
3131
param_names = ['dtype']
3232

3333
def setup(self, dtype):
3434
self.s = Series(np.random.randint(1, 10, 100000)).astype(dtype)
3535
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)
3639

3740
def time_isin(self, dtypes):
3841
self.s.isin(self.values)
3942

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+
4051

4152
class NSort(object):
4253

0 commit comments

Comments
 (0)