17
17
18
18
19
19
class TestCategoricalIndex (Base ):
20
- _holder = CategoricalIndex
20
+ _index_cls = CategoricalIndex
21
+
22
+ @pytest .fixture
23
+ def simple_index (self ) -> CategoricalIndex :
24
+ return self ._index_cls (list ("aabbca" ), categories = list ("cab" ), ordered = False )
21
25
22
26
@pytest .fixture
23
27
def index (self , request ):
24
28
return tm .makeCategoricalIndex (100 )
25
29
26
- def create_index (self , categories = None , ordered = False ):
30
+ def create_index (self , * , categories = None , ordered = False ):
27
31
if categories is None :
28
32
categories = list ("cab" )
29
33
return CategoricalIndex (list ("aabbca" ), categories = categories , ordered = ordered )
@@ -33,9 +37,9 @@ def test_can_hold_identifiers(self):
33
37
key = idx [0 ]
34
38
assert idx ._can_hold_identifiers_and_holds_name (key ) is True
35
39
36
- def test_insert (self ):
40
+ def test_insert (self , simple_index ):
37
41
38
- ci = self . create_index ()
42
+ ci = simple_index
39
43
categories = ci .categories
40
44
41
45
# test 0th element
@@ -70,9 +74,9 @@ def test_insert_na_mismatched_dtype(self):
70
74
expected = Index ([pd .NaT , 0 , 1 , 1 ], dtype = object )
71
75
tm .assert_index_equal (result , expected )
72
76
73
- def test_delete (self ):
77
+ def test_delete (self , simple_index ):
74
78
75
- ci = self . create_index ()
79
+ ci = simple_index
76
80
categories = ci .categories
77
81
78
82
result = ci .delete (0 )
0 commit comments