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