Skip to content

Commit 7424d2e

Browse files
committed
test more mask types for putmask
1 parent 990ce3b commit 7424d2e

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/indexes/test_indexes.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1589,6 +1589,12 @@ def test_index_setitem() -> None:
15891589
def test_index_putmask() -> None:
15901590
idx = pd.Index([1, 2])
15911591
check(assert_type(idx.putmask([True, False], 11.4), "pd.Index"), pd.Index)
1592+
check(assert_type(idx.putmask(np.array([True, False]), 11.4), "pd.Index"), pd.Index)
1593+
check(
1594+
assert_type(idx.putmask(pd.Series([True, False]), 11.4), "pd.Index"), pd.Index
1595+
)
1596+
check(assert_type(idx.putmask(pd.Index([True, False]), 11.4), "pd.Index"), pd.Index)
1597+
check(assert_type(idx.putmask(pd.array([True, False]), 11.5), "pd.Index"), pd.Index)
15921598

15931599

15941600
def test_index_asof() -> None:

0 commit comments

Comments
 (0)