Skip to content

Commit 2255ce4

Browse files
committed
Add >1 length mask test
1 parent 96f85ac commit 2255ce4

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pandas/tests/indexes/ranges/test_range.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -716,6 +716,18 @@ def test__getitem__boolean_numpyextensionarray():
716716
tm.assert_index_equal(ri, result)
717717

718718

719+
@pytest.mark.parametrize(
720+
"container",
721+
[np.array, pd.Series, lambda x: pd.arrays.NumpyExtensionArray(np.array(x))],
722+
ids=["numpy-array", "series", "numpy-extension-array"],
723+
)
724+
def test__getitem__boolean_arraylike(container):
725+
ri = RangeIndex(5)
726+
result = ri[container([True, True, False, False, True])]
727+
expected = Index([0, 1, 4], dtype="int64")
728+
tm.assert_index_equal(result, expected)
729+
730+
719731
@pytest.mark.parametrize(
720732
"rng, exp_rng",
721733
[

0 commit comments

Comments
 (0)