From 9cfedf48fa79e7cf42643069cf5f12e11cdc2399 Mon Sep 17 00:00:00 2001 From: Michael Tiemann <72577720+MichaelTiemannOSC@users.noreply.github.com> Date: Mon, 7 Aug 2023 09:02:32 -0400 Subject: [PATCH] test_map: use data_missing to exercise na_action Change `data` to `data_missing` to exercise the `na_action` parameter. Signed-off-by: Michael Tiemann <72577720+MichaelTiemannOSC@users.noreply.github.com> --- pandas/tests/extension/base/methods.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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):