@@ -1447,20 +1447,36 @@ def test_to_hierarchical(self):
1447
1447
def test_bounds (self ):
1448
1448
self .index ._bounds
1449
1449
1450
+ def test_equals (self ):
1451
+
1452
+ for name , idx in compat .iteritems (self .indices ):
1453
+ self .assertTrue (idx .equals (idx ))
1454
+ self .assertTrue (idx .equals (idx .copy ()))
1455
+ self .assertTrue (idx .equals (idx .astype (object )))
1456
+ self .assertTrue (idx .equals (list (idx )))
1457
+ self .assertTrue (idx .equals (np .array (idx )))
1458
+
1459
+ same_values = Index (idx , dtype = object )
1460
+ self .assertTrue (idx .equals (same_values ))
1461
+ self .assertTrue (same_values .equals (idx ))
1462
+
1450
1463
def test_equals_multi (self ):
1451
- self .assertTrue (self .index .equals (self .index ))
1452
- self .assertTrue (self .index .equal_levels (self .index ))
1464
+ assert self .index .equals (self .index )
1465
+ assert self .index .equals (self .index .values )
1466
+ assert self .index .equals (Index (self .index .values ))
1453
1467
1454
- self .assertFalse (self .index .equals (self .index [:- 1 ]))
1468
+ assert self .index .equal_levels (self .index )
1469
+ assert not self .index .equals (self .index [:- 1 ])
1470
+ assert not self .index .equals (self .index [- 1 ])
1455
1471
1456
1472
# different number of levels
1457
1473
index = MultiIndex (levels = [Index (lrange (4 )), Index (lrange (4 )), Index (
1458
1474
lrange (4 ))], labels = [np .array ([0 , 0 , 1 , 2 , 2 , 2 , 3 , 3 ]), np .array (
1459
1475
[0 , 1 , 0 , 0 , 0 , 1 , 0 , 1 ]), np .array ([1 , 0 , 1 , 1 , 0 , 0 , 1 , 0 ])])
1460
1476
1461
1477
index2 = MultiIndex (levels = index .levels [:- 1 ], labels = index .labels [:- 1 ])
1462
- self . assertFalse ( index .equals (index2 ) )
1463
- self . assertFalse ( index .equal_levels (index2 ) )
1478
+ assert not index .equals (index2 )
1479
+ assert not index .equal_levels (index2 )
1464
1480
1465
1481
# levels are different
1466
1482
major_axis = Index (lrange (4 ))
@@ -1471,8 +1487,8 @@ def test_equals_multi(self):
1471
1487
1472
1488
index = MultiIndex (levels = [major_axis , minor_axis ],
1473
1489
labels = [major_labels , minor_labels ])
1474
- self .assertFalse ( self . index .equals (index ) )
1475
- self .assertFalse ( self . index .equal_levels (index ) )
1490
+ assert not self .index .equals (index )
1491
+ assert not self .index .equal_levels (index )
1476
1492
1477
1493
# some of the labels are different
1478
1494
major_axis = Index (['foo' , 'bar' , 'baz' , 'qux' ])
@@ -1483,7 +1499,7 @@ def test_equals_multi(self):
1483
1499
1484
1500
index = MultiIndex (levels = [major_axis , minor_axis ],
1485
1501
labels = [major_labels , minor_labels ])
1486
- self .assertFalse ( self . index .equals (index ) )
1502
+ assert not self .index .equals (index )
1487
1503
1488
1504
def test_equals_missing_values (self ):
1489
1505
# make sure take is not using -1
0 commit comments