Skip to content

Commit 7c5d3d5

Browse files
authored
REF: include CategoricalIndex in index_cached parametrization (#32215)
1 parent 89ed095 commit 7c5d3d5

File tree

7 files changed

+11
-17
lines changed

7 files changed

+11
-17
lines changed

asv_bench/benchmarks/categoricals.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -258,9 +258,6 @@ def setup(self):
258258
def time_get_loc(self):
259259
self.index.get_loc(self.category)
260260

261-
def time_shape(self):
262-
self.index.shape
263-
264261
def time_shallow_copy(self):
265262
self.index._shallow_copy()
266263

asv_bench/benchmarks/index_cached_properties.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ class IndexCache:
77

88
params = [
99
[
10+
"CategoricalIndex",
1011
"DatetimeIndex",
1112
"Float64Index",
1213
"IntervalIndex",
@@ -42,6 +43,8 @@ def setup(self, index_type):
4243
self.idx = pd.Float64Index(range(N))
4344
elif index_type == "UInt64Index":
4445
self.idx = pd.UInt64Index(range(N))
46+
elif index_type == "CategoricalIndex":
47+
self.idx = pd.CategoricalIndex(range(N), range(N))
4548
else:
4649
raise ValueError
4750
assert len(self.idx) == N

asv_bench/benchmarks/index_object.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -55,14 +55,6 @@ def time_datetime_difference_disjoint(self):
5555
self.datetime_left.difference(self.datetime_right)
5656

5757

58-
class Datetime:
59-
def setup(self):
60-
self.dr = date_range("20000101", freq="D", periods=10000)
61-
62-
def time_is_dates_only(self):
63-
self.dr._is_dates_only
64-
65-
6658
class Range:
6759
def setup(self):
6860
self.idx_inc = RangeIndex(start=0, stop=10 ** 7, step=3)

asv_bench/benchmarks/indexing.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,8 @@
1+
"""
2+
These benchmarks are for Series and DataFrame indexing methods. For the
3+
lower-level methods directly on Index and subclasses, see index_object.py,
4+
indexing_engine.py, and index_cached.py
5+
"""
16
import warnings
27

38
import numpy as np

asv_bench/benchmarks/period.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,9 +85,6 @@ def setup(self):
8585
def time_get_loc(self):
8686
self.index.get_loc(self.period)
8787

88-
def time_shape(self):
89-
self.index.shape
90-
9188
def time_shallow_copy(self):
9289
self.index._shallow_copy()
9390

asv_bench/benchmarks/timedelta.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,9 +73,6 @@ def setup(self):
7373
def time_get_loc(self):
7474
self.index.get_loc(self.timedelta)
7575

76-
def time_shape(self):
77-
self.index.shape
78-
7976
def time_shallow_copy(self):
8077
self.index._shallow_copy()
8178

asv_bench/benchmarks/timeseries.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,9 @@ def time_to_date(self, index_type):
5757
def time_to_pydatetime(self, index_type):
5858
self.index.to_pydatetime()
5959

60+
def time_is_dates_only(self, index_type):
61+
self.index._is_dates_only
62+
6063

6164
class TzLocalize:
6265

0 commit comments

Comments
 (0)