diff --git a/pandas/tests/extension/base/methods.py b/pandas/tests/extension/base/methods.py index 2dd62a4ca7538..e696e1881bf74 100644 --- a/pandas/tests/extension/base/methods.py +++ b/pandas/tests/extension/base/methods.py @@ -94,9 +94,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_missing, na_action): - result = data_missing.map(lambda x: x, na_action=na_action) - expected = data_missing.to_numpy() + def test_map(self, data, na_action): + result = data.map(lambda x: x, na_action=na_action) + expected = data.to_numpy() tm.assert_numpy_array_equal(result, expected) def test_argsort(self, data_for_sorting): @@ -697,4 +697,4 @@ def test_equals(self, data, na_value, as_series, box): def test_equals_same_data_different_object(self, data): # https://github.com/pandas-dev/pandas/issues/34660 - assert pd.Series(data).equals(pd.Series(data)) + assert pd.Series(data).equals(pd.Series(data)) \ No newline at end of file