From d610ab34318aac6d9c56cf11ce9cb5122cd50bb1 Mon Sep 17 00:00:00 2001 From: Dawid Makar Date: Tue, 26 Sep 2023 23:23:55 +0200 Subject: [PATCH] Automatic refactoring. Refactoring step id: UUID('cb1c9915-7432-4024-944e-bfff5f59124a') --- pandas/tests/extension/base/methods.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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