diff --git a/pandas/tests/frame/indexing/test_mask.py b/pandas/tests/frame/indexing/test_mask.py index ac6f0a1ac0f73..e4036efeab7ff 100644 --- a/pandas/tests/frame/indexing/test_mask.py +++ b/pandas/tests/frame/indexing/test_mask.py @@ -105,7 +105,7 @@ def test_mask_stringdtype(frame_or_series): {"A": ["this", "that"]}, index=["id2", "id3"], dtype=StringDtype() ) expected = DataFrame( - {"A": [NA, "this", "that", NA]}, + {"A": ["foo", "this", "that", NA]}, index=["id1", "id2", "id3", "id4"], dtype=StringDtype(), ) @@ -114,7 +114,10 @@ def test_mask_stringdtype(frame_or_series): filtered_obj = filtered_obj["A"] expected = expected["A"] - filter_ser = Series([False, True, True, False]) + filter_ser = Series( + [False, True, True, False], + index=["id1", "id2", "id3", "id4"], + ) result = obj.mask(filter_ser, filtered_obj) tm.assert_equal(result, expected)