Skip to content

Commit 688e2a0

Browse files
authored
TST: Fix test_mask_stringdtype (#61830)
Fix test
1 parent 337d5fe commit 688e2a0

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

pandas/tests/frame/indexing/test_mask.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ def test_mask_stringdtype(frame_or_series):
105105
{"A": ["this", "that"]}, index=["id2", "id3"], dtype=StringDtype()
106106
)
107107
expected = DataFrame(
108-
{"A": [NA, "this", "that", NA]},
108+
{"A": ["foo", "this", "that", NA]},
109109
index=["id1", "id2", "id3", "id4"],
110110
dtype=StringDtype(),
111111
)
@@ -114,7 +114,10 @@ def test_mask_stringdtype(frame_or_series):
114114
filtered_obj = filtered_obj["A"]
115115
expected = expected["A"]
116116

117-
filter_ser = Series([False, True, True, False])
117+
filter_ser = Series(
118+
[False, True, True, False],
119+
index=["id1", "id2", "id3", "id4"],
120+
)
118121
result = obj.mask(filter_ser, filtered_obj)
119122

120123
tm.assert_equal(result, expected)

0 commit comments

Comments
 (0)