@@ -75,50 +75,45 @@ def test_searchsorted(self, ordered):
75
75
# https://github.com/pandas-dev/pandas/issues/8420
76
76
# https://github.com/pandas-dev/pandas/issues/14522
77
77
78
- c = Categorical (['cheese' , 'milk' , 'apple' , 'bread' , 'bread' ],
79
- categories = ['cheese' , 'milk' , 'apple' , 'bread' ],
80
- ordered = ordered )
81
- s = Series (c )
78
+ cat = Categorical (['cheese' , 'milk' , 'apple' , 'bread' , 'bread' ],
79
+ categories = ['cheese' , 'milk' , 'apple' , 'bread' ],
80
+ ordered = ordered )
81
+ ser = Series (cat )
82
82
83
83
# Searching for single item argument, side='left' (default)
84
- res_cat = c .searchsorted ('apple' )
84
+ res_cat = cat .searchsorted ('apple' )
85
85
assert res_cat == 2
86
86
assert is_scalar (res_cat )
87
87
88
- res_ser = s .searchsorted ('apple' )
88
+ res_ser = ser .searchsorted ('apple' )
89
89
assert res_ser == 2
90
90
assert is_scalar (res_ser )
91
91
92
- res_cat = c .searchsorted ('apple' )
93
- res_ser = s .searchsorted ('apple' )
94
- assert res_cat == 2
95
- assert res_ser == 2
96
-
97
92
# Searching for single item array, side='left' (default)
98
- res_cat = c .searchsorted (['bread' ])
99
- res_ser = s .searchsorted (['bread' ])
93
+ res_cat = cat .searchsorted (['bread' ])
94
+ res_ser = ser .searchsorted (['bread' ])
100
95
exp = np .array ([3 ], dtype = np .intp )
101
96
tm .assert_numpy_array_equal (res_cat , exp )
102
97
tm .assert_numpy_array_equal (res_ser , exp )
103
98
104
99
# Searching for several items array, side='right'
105
- res_cat = c .searchsorted (['apple' , 'bread' ], side = 'right' )
106
- res_ser = s .searchsorted (['apple' , 'bread' ], side = 'right' )
100
+ res_cat = cat .searchsorted (['apple' , 'bread' ], side = 'right' )
101
+ res_ser = ser .searchsorted (['apple' , 'bread' ], side = 'right' )
107
102
exp = np .array ([3 , 5 ], dtype = np .intp )
108
103
tm .assert_numpy_array_equal (res_cat , exp )
109
104
tm .assert_numpy_array_equal (res_ser , exp )
110
105
111
106
# Searching for a single value that is not from the Categorical
112
107
with pytest .raises (KeyError ):
113
- c .searchsorted ('cucumber' )
108
+ cat .searchsorted ('cucumber' )
114
109
with pytest .raises (KeyError ):
115
- s .searchsorted ('cucumber' )
110
+ ser .searchsorted ('cucumber' )
116
111
117
112
# Searching for multiple values one of each is not from the Categorical
118
113
with pytest .raises (KeyError ):
119
- c .searchsorted (['bread' , 'cucumber' ])
114
+ cat .searchsorted (['bread' , 'cucumber' ])
120
115
with pytest .raises (KeyError ):
121
- s .searchsorted (['bread' , 'cucumber' ])
116
+ ser .searchsorted (['bread' , 'cucumber' ])
122
117
123
118
def test_unique (self ):
124
119
# categories are reordered based on value when ordered=False
0 commit comments