Skip to content

Commit c6d23f6

Browse files
Daniel SaxtonDaniel Saxton
Daniel Saxton
authored and
Daniel Saxton
committed
Add setitem test
1 parent a6294f8 commit c6d23f6

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pandas/tests/extension/base/setitem.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,18 @@ def test_setitem_mask_raises(self, data, box_in_series):
125125
with pytest.raises(IndexError, match="wrong length"):
126126
data[mask] = data[0]
127127

128+
def test_setitem_mask_boolean_array_with_na(self, data, box_in_series):
129+
mask = pd.array(np.zeros(data.shape, dtype="bool"), dtype="boolean")
130+
mask[:3] = True
131+
mask[3:5] = pd.NA
132+
133+
if box_in_series:
134+
data = pd.Series(data)
135+
136+
data[mask] = data[0]
137+
138+
assert (data[:3] == data[0]).all()
139+
128140
@pytest.mark.parametrize(
129141
"idx",
130142
[[0, 1, 2], pd.array([0, 1, 2], dtype="Int64"), np.array([0, 1, 2])],

0 commit comments

Comments
 (0)