@@ -78,35 +78,30 @@ def test_searchsorted(self, ordered):
78
78
# https://github.com/pandas-dev/pandas/issues/8420
79
79
# https://github.com/pandas-dev/pandas/issues/14522
80
80
81
- c = Categorical (['cheese' , 'milk' , 'apple' , 'bread' , 'bread' ],
82
- categories = ['cheese' , 'milk' , 'apple' , 'bread' ],
83
- ordered = ordered )
84
- s = Series (c )
81
+ cat = Categorical (['cheese' , 'milk' , 'apple' , 'bread' , 'bread' ],
82
+ categories = ['cheese' , 'milk' , 'apple' , 'bread' ],
83
+ ordered = ordered )
84
+ ser = Series (cat )
85
85
86
86
# Searching for single item argument, side='left' (default)
87
- res_cat = c .searchsorted ('apple' )
87
+ res_cat = cat .searchsorted ('apple' )
88
88
assert res_cat == 2
89
89
assert is_scalar (res_cat )
90
90
91
- res_ser = s .searchsorted ('apple' )
91
+ res_ser = ser .searchsorted ('apple' )
92
92
assert res_ser == 2
93
93
assert is_scalar (res_ser )
94
94
95
- res_cat = c .searchsorted ('apple' )
96
- res_ser = s .searchsorted ('apple' )
97
- assert res_cat == 2
98
- assert res_ser == 2
99
-
100
95
# Searching for single item array, side='left' (default)
101
- res_cat = c .searchsorted (['bread' ])
102
- res_ser = s .searchsorted (['bread' ])
96
+ res_cat = cat .searchsorted (['bread' ])
97
+ res_ser = ser .searchsorted (['bread' ])
103
98
exp = np .array ([3 ], dtype = np .intp )
104
99
tm .assert_numpy_array_equal (res_cat , exp )
105
100
tm .assert_numpy_array_equal (res_ser , exp )
106
101
107
102
# Searching for several items array, side='right'
108
- res_cat = c .searchsorted (['apple' , 'bread' ], side = 'right' )
109
- res_ser = s .searchsorted (['apple' , 'bread' ], side = 'right' )
103
+ res_cat = cat .searchsorted (['apple' , 'bread' ], side = 'right' )
104
+ res_ser = ser .searchsorted (['apple' , 'bread' ], side = 'right' )
110
105
exp = np .array ([3 , 5 ], dtype = np .intp )
111
106
tm .assert_numpy_array_equal (res_cat , exp )
112
107
tm .assert_numpy_array_equal (res_ser , exp )
0 commit comments