Skip to content

Commit d7fc3b7

Browse files
Daniel SaxtonDaniel Saxton
Daniel Saxton
authored and
Daniel Saxton
committed
Revert "Include list mask in test"
This reverts commit a559385.
1 parent a10765f commit d7fc3b7

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

pandas/tests/extension/base/getitem.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -158,26 +158,20 @@ def test_getitem_boolean_array_mask(self, data):
158158
result = pd.Series(data)[mask]
159159
self.assert_series_equal(result, expected)
160160

161-
@pytest.mark.parametrize("use_list_mask", [True, False])
162-
def test_getitem_boolean_na_treated_as_false(self, data, use_list_mask):
161+
def test_getitem_boolean_na_treated_as_false(self, data):
163162
mask = pd.array(np.zeros(data.shape, dtype="bool"), dtype="boolean")
164163
mask[:2] = pd.NA
165164
mask[2:4] = True
166-
filled_mask = mask.fillna(False)
167-
168-
if use_list_mask:
169-
mask = list(mask)
170-
filled_mask = list(filled_mask)
171165

172166
result = data[mask]
173-
expected = data[filled_mask]
167+
expected = data[mask.fillna(False)]
174168

175169
self.assert_extension_array_equal(result, expected)
176170

177171
s = pd.Series(data)
178172

179173
result = s[mask]
180-
expected = s[filled_mask]
174+
expected = s[mask.fillna(False)]
181175

182176
self.assert_series_equal(result, expected)
183177

0 commit comments

Comments
 (0)