Skip to content

Commit adac8f4

Browse files
committed
clean up ASV code a bit
1 parent 67fae0e commit adac8f4

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

asv_bench/benchmarks/indexing_engines.py

+2-5
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
import numpy as np
22

3-
import pandas.util.testing as tm
43
from pandas._libs.index import (Int64Engine, Int32Engine,
54
Int16Engine, Int8Engine,
65
UInt64Engine, UInt32Engine,
@@ -32,10 +31,9 @@ def setup(self, engine, index_type):
3231
}[index_type]
3332

3433
self.data = engine(lambda: array_, len(array_))
35-
self.int_scalar = 2
3634

3735
def time_get_loc(self, engine, index_type):
38-
self.data.get_loc(self.int_scalar)
36+
self.data.get_loc(2)
3937

4038

4139
class ObjectEngineIndexing(object):
@@ -57,7 +55,6 @@ def setup(self, engine, index_type):
5755
}[index_type]
5856

5957
self.data = engine(lambda: array_, len(array_))
60-
self.int_scalar = 'b'
6158

6259
def time_get_loc(self, engine, index_type):
63-
self.data.get_loc(self.int_scalar)
60+
self.data.get_loc(2)

0 commit comments

Comments
 (0)