@@ -149,13 +149,16 @@ def test_set_levels(self):
149
149
levels = self .index .levels
150
150
new_levels = [[lev + 'a' for lev in level ] for level in levels ]
151
151
152
- def assert_matching (actual , expected ):
152
+ def assert_matching (actual , expected , coerce_to_obj = True ):
153
153
# avoid specifying internal representation
154
154
# as much as possible
155
155
self .assertEqual (len (actual ), len (expected ))
156
156
for act , exp in zip (actual , expected ):
157
- act = np .asarray (act , dtype = np .object_ )
158
- exp = np .asarray (exp , dtype = np .object_ )
157
+ act = np .asarray (act )
158
+ if coerce_to_obj :
159
+ exp = np .asarray (exp , dtype = np .object_ )
160
+ else :
161
+ exp = np .asarray (exp )
159
162
tm .assert_numpy_array_equal (act , exp )
160
163
161
164
# level changing [w/o mutation]
@@ -210,20 +213,24 @@ def assert_matching(actual, expected):
210
213
for inplace in [True , False ]:
211
214
with assertRaisesRegexp (ValueError , "^On" ):
212
215
self .index .set_levels (['c' ], level = 0 , inplace = inplace )
213
- assert_matching (self .index .levels , original_index .levels )
216
+ assert_matching (self .index .levels , original_index .levels ,
217
+ coerce_to_obj = False )
214
218
215
219
with assertRaisesRegexp (ValueError , "^On" ):
216
220
self .index .set_labels ([0 , 1 , 2 , 3 , 4 , 5 ], level = 0 ,
217
221
inplace = inplace )
218
- assert_matching (self .index .labels , original_index .labels )
222
+ assert_matching (self .index .labels , original_index .labels ,
223
+ coerce_to_obj = False )
219
224
220
225
with assertRaisesRegexp (TypeError , "^Levels" ):
221
226
self .index .set_levels ('c' , level = 0 , inplace = inplace )
222
- assert_matching (self .index .levels , original_index .levels )
227
+ assert_matching (self .index .levels , original_index .levels ,
228
+ coerce_to_obj = False )
223
229
224
230
with assertRaisesRegexp (TypeError , "^Labels" ):
225
231
self .index .set_labels (1 , level = 0 , inplace = inplace )
226
- assert_matching (self .index .labels , original_index .labels )
232
+ assert_matching (self .index .labels , original_index .labels ,
233
+ coerce_to_obj = False )
227
234
228
235
def test_set_labels (self ):
229
236
# side note - you probably wouldn't want to use levels and labels
0 commit comments