Skip to content

Commit 0f558aa

Browse files
committed
TST: MultiIndex comparison with tuple pandas-dev#21517
1 parent 640fe50 commit 0f558aa

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

pandas/tests/indexes/multi/test_equivalence.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,32 @@ def test_equals_op(idx):
8484
tm.assert_series_equal(series_a == item, Series(expected3))
8585

8686

87+
def test_compare_tuple():
88+
# GH#21517
89+
mi = MultiIndex.from_product([[1, 2]] * 2)
90+
91+
all_false = np.array([False, False, False, False])
92+
93+
result = mi == mi[0]
94+
expected = np.array([True, False, False, False])
95+
tm.assert_numpy_array_equal(result, expected)
96+
97+
result = mi != mi[0]
98+
tm.assert_numpy_array_equal(result, ~expected)
99+
100+
result = mi < mi[0]
101+
tm.assert_numpy_array_equal(result, all_false)
102+
103+
result = mi <= mi[0]
104+
tm.assert_numpy_array_equal(result, expected)
105+
106+
result = mi > mi[0]
107+
tm.assert_numpy_array_equal(result, ~expected)
108+
109+
result = mi >= mi[0]
110+
tm.assert_numpy_array_equal(result, ~all_false)
111+
112+
87113
def test_equals_multi(idx):
88114
assert idx.equals(idx)
89115
assert not idx.equals(idx.values)

0 commit comments

Comments
 (0)