From 9cdc7686aad94be797cf4dc605ca122cddbe5c7a Mon Sep 17 00:00:00 2001 From: arthurlw Date: Fri, 11 Jul 2025 10:32:14 +0700 Subject: [PATCH] Fix test --- pandas/tests/frame/indexing/test_mask.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) 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)