@@ -1448,19 +1448,21 @@ def test_bounds(self):
1448
1448
self .index ._bounds
1449
1449
1450
1450
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 )
1453
1453
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 ])
1455
1457
1456
1458
# different number of levels
1457
1459
index = MultiIndex (levels = [Index (lrange (4 )), Index (lrange (4 )), Index (
1458
1460
lrange (4 ))], labels = [np .array ([0 , 0 , 1 , 2 , 2 , 2 , 3 , 3 ]), np .array (
1459
1461
[0 , 1 , 0 , 0 , 0 , 1 , 0 , 1 ]), np .array ([1 , 0 , 1 , 1 , 0 , 0 , 1 , 0 ])])
1460
1462
1461
1463
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 )
1464
1466
1465
1467
# levels are different
1466
1468
major_axis = Index (lrange (4 ))
@@ -1471,8 +1473,8 @@ def test_equals_multi(self):
1471
1473
1472
1474
index = MultiIndex (levels = [major_axis , minor_axis ],
1473
1475
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 )
1476
1478
1477
1479
# some of the labels are different
1478
1480
major_axis = Index (['foo' , 'bar' , 'baz' , 'qux' ])
@@ -1483,7 +1485,7 @@ def test_equals_multi(self):
1483
1485
1484
1486
index = MultiIndex (levels = [major_axis , minor_axis ],
1485
1487
labels = [major_labels , minor_labels ])
1486
- self .assertFalse ( self . index .equals (index ) )
1488
+ assert not self .index .equals (index )
1487
1489
1488
1490
def test_equals_missing_values (self ):
1489
1491
# make sure take is not using -1
0 commit comments