Skip to content

Commit 463f527

Browse files
committed
more fixes
1 parent 6db38b8 commit 463f527

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

pandas/tests/indexes/test_multi.py

+10-8
Original file line numberDiff line numberDiff line change
@@ -1448,19 +1448,21 @@ def test_bounds(self):
14481448
self.index._bounds
14491449

14501450
def test_equals_multi(self):
1451-
self.assertTrue(self.index.equals(self.index))
1452-
self.assertTrue(self.index.equal_levels(self.index))
1451+
assert self.index.equals(self.index)
1452+
assert not self.index.equals(self.index.values)
14531453

1454-
self.assertFalse(self.index.equals(self.index[:-1]))
1454+
assert self.index.equal_levels(self.index)
1455+
assert not self.index.equals(self.index[:-1])
1456+
assert not self.index.equals(self.index[-1])
14551457

14561458
# different number of levels
14571459
index = MultiIndex(levels=[Index(lrange(4)), Index(lrange(4)), Index(
14581460
lrange(4))], labels=[np.array([0, 0, 1, 2, 2, 2, 3, 3]), np.array(
14591461
[0, 1, 0, 0, 0, 1, 0, 1]), np.array([1, 0, 1, 1, 0, 0, 1, 0])])
14601462

14611463
index2 = MultiIndex(levels=index.levels[:-1], labels=index.labels[:-1])
1462-
self.assertFalse(index.equals(index2))
1463-
self.assertFalse(index.equal_levels(index2))
1464+
assert not index.equals(index2)
1465+
assert not index.equal_levels(index2)
14641466

14651467
# levels are different
14661468
major_axis = Index(lrange(4))
@@ -1471,8 +1473,8 @@ def test_equals_multi(self):
14711473

14721474
index = MultiIndex(levels=[major_axis, minor_axis],
14731475
labels=[major_labels, minor_labels])
1474-
self.assertFalse(self.index.equals(index))
1475-
self.assertFalse(self.index.equal_levels(index))
1476+
assert not self.index.equals(index)
1477+
assert not self.index.equal_levels(index)
14761478

14771479
# some of the labels are different
14781480
major_axis = Index(['foo', 'bar', 'baz', 'qux'])
@@ -1483,7 +1485,7 @@ def test_equals_multi(self):
14831485

14841486
index = MultiIndex(levels=[major_axis, minor_axis],
14851487
labels=[major_labels, minor_labels])
1486-
self.assertFalse(self.index.equals(index))
1488+
assert not self.index.equals(index)
14871489

14881490
def test_equals_missing_values(self):
14891491
# make sure take is not using -1

0 commit comments

Comments
 (0)