File tree Expand file tree Collapse file tree 1 file changed +3
-9
lines changed
pandas/tests/extension/base Expand file tree Collapse file tree 1 file changed +3
-9
lines changed Original file line number Diff line number Diff line change @@ -158,26 +158,20 @@ def test_getitem_boolean_array_mask(self, data):
158
158
result = pd .Series (data )[mask ]
159
159
self .assert_series_equal (result , expected )
160
160
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 ):
163
162
mask = pd .array (np .zeros (data .shape , dtype = "bool" ), dtype = "boolean" )
164
163
mask [:2 ] = pd .NA
165
164
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 )
171
165
172
166
result = data [mask ]
173
- expected = data [filled_mask ]
167
+ expected = data [mask . fillna ( False ) ]
174
168
175
169
self .assert_extension_array_equal (result , expected )
176
170
177
171
s = pd .Series (data )
178
172
179
173
result = s [mask ]
180
- expected = s [filled_mask ]
174
+ expected = s [mask . fillna ( False ) ]
181
175
182
176
self .assert_series_equal (result , expected )
183
177
You can’t perform that action at this time.
0 commit comments