Skip to content

Commit 005ab51

Browse files
author
Ben Kandel
committed
fix test coercion
1 parent e1a6e0a commit 005ab51

File tree

1 file changed

+14
-7
lines changed

1 file changed

+14
-7
lines changed

pandas/tests/indexes/test_multi.py

+14-7
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,16 @@ def test_set_levels(self):
149149
levels = self.index.levels
150150
new_levels = [[lev + 'a' for lev in level] for level in levels]
151151

152-
def assert_matching(actual, expected):
152+
def assert_matching(actual, expected, coerce_to_obj=True):
153153
# avoid specifying internal representation
154154
# as much as possible
155155
self.assertEqual(len(actual), len(expected))
156156
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)
159162
tm.assert_numpy_array_equal(act, exp)
160163

161164
# level changing [w/o mutation]
@@ -210,20 +213,24 @@ def assert_matching(actual, expected):
210213
for inplace in [True, False]:
211214
with assertRaisesRegexp(ValueError, "^On"):
212215
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)
214218

215219
with assertRaisesRegexp(ValueError, "^On"):
216220
self.index.set_labels([0, 1, 2, 3, 4, 5], level=0,
217221
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)
219224

220225
with assertRaisesRegexp(TypeError, "^Levels"):
221226
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)
223229

224230
with assertRaisesRegexp(TypeError, "^Labels"):
225231
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)
227234

228235
def test_set_labels(self):
229236
# side note - you probably wouldn't want to use levels and labels

0 commit comments

Comments
 (0)