diff --git a/pandas/tests/extension/base/methods.py b/pandas/tests/extension/base/methods.py index dea43e53ffc40..93c8e873e0bc3 100644 --- a/pandas/tests/extension/base/methods.py +++ b/pandas/tests/extension/base/methods.py @@ -95,9 +95,9 @@ def test_apply_simple_series(self, data): assert isinstance(result, pd.Series) @pytest.mark.parametrize("na_action", [None, "ignore"]) - def test_map(self, data, na_action): - result = data.map(lambda x: x, na_action=na_action) - expected = data.to_numpy() + def test_map(self, data_missing, na_action): + result = data_missing.map(lambda x: x, na_action=na_action) + expected = data_missing.to_numpy() tm.assert_numpy_array_equal(result, expected) def test_argsort(self, data_for_sorting):