Skip to content

Commit cd7fd6b

Browse files
committed
TST: asv tests for indexing in non-unique MultiIndex
1 parent 7463825 commit cd7fd6b

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

asv_bench/benchmarks/multiindex_object.py

+8
Original file line numberDiff line numberDiff line change
@@ -91,10 +91,18 @@ def setup(self):
9191
1000 + np.arange(n)]
9292
labels = [np.random.choice(n, (k * n)) for lev in levels]
9393
self.mi = MultiIndex(levels=levels, labels=labels)
94+
self.key = self.mi[len(self.mi) // 2]
95+
self.partial_key = (self.key[0], self.key[1])
9496

9597
def time_duplicated(self):
9698
self.mi.duplicated()
9799

100+
def time_duplicated_loc(self):
101+
self.mi.get_loc(self.key)
102+
103+
def time_duplicated_partial_loc(self):
104+
self.mi.get_loc(self.partial_key)
105+
98106

99107
class Sortlevel(object):
100108

0 commit comments

Comments
 (0)