Skip to content

Commit 8ac278b

Browse files
committed
PERF: expand UInt64Index benchmark coverage
1 parent 385b052 commit 8ac278b

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

asv_bench/benchmarks/algorithms.py

+4
Original file line numberDiff line numberDiff line change
@@ -43,12 +43,16 @@ class Duplicated(object):
4343
def setup(self, keep):
4444
N = 10**5
4545
self.int_idx = pd.Int64Index(np.arange(N).repeat(5))
46+
self.uint_idx = pd.UInt64Index(np.arange(N).repeat(5))
4647
self.float_idx = pd.Float64Index(np.random.randn(N).repeat(5))
4748
self.string_idx = tm.makeStringIndex(N)
4849

4950
def time_duplicated_int(self, keep):
5051
self.int_idx.duplicated(keep=keep)
5152

53+
def time_duplicated_uint(self, keep):
54+
self.uint_idx.duplicated(keep=keep)
55+
5256
def time_duplicated_float(self, keep):
5357
self.float_idx.duplicated(keep=keep)
5458

asv_bench/benchmarks/series_methods.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def time_constructor(self, data):
2323

2424
class IsIn(object):
2525

26-
params = ['int64', 'object']
26+
params = ['int64', 'uint64', 'object']
2727
param_names = ['dtype']
2828

2929
def setup(self, dtype):
@@ -150,7 +150,7 @@ def time_clip(self):
150150

151151
class ValueCounts(object):
152152

153-
params = ['int', 'float', 'object']
153+
params = ['int', 'uint', 'float', 'object']
154154
param_names = ['dtype']
155155

156156
def setup(self, dtype):

0 commit comments

Comments
 (0)