Skip to content

Commit c073c0b

Browse files
committed
Add test for list and ndarray
1 parent 9574746 commit c073c0b

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

pandas/tests/series/indexing/test_mask.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,14 @@ def test_mask_inplace():
8181
def test_mask_na(dtype):
8282
# We should not be filling pd.NA. See GH#60729
8383
series = Series([None, 1, 2, None, 3, 4, None], dtype=dtype)
84-
result = series.mask(series <= 2, -99)
84+
cond = series <= 2
8585
expected = Series([None, -99, -99, None, 3, 4, None], dtype=dtype)
8686

87+
result = series.mask(cond, -99)
88+
tm.assert_series_equal(result, expected)
89+
90+
result = series.mask(cond.to_list(), expected)
91+
tm.assert_series_equal(result, expected)
92+
93+
result = series.mask(cond.to_numpy(), expected)
8794
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)