Update na_action for dataframe map

Author: kopytjuk

pandas/conftest.py

@@ -405,7 +405,15 @@ def nselect_method(request):
 @pytest.fixture(params=[None, "ignore", "raise"])
 def na_action(request):
     """
-    Fixture for 'na_action' argument in map.
+    Fixture for 'na_action' argument in Series.map.
+    """
+    return request.param
+
+
+@pytest.fixture(params=[None, "ignore"])
+def na_action_frame(request):
+    """
+    Fixture for 'na_action' argument in DataFrame.map.
     """
     return request.param
 

pandas/tests/frame/methods/test_map.py

@@ -33,7 +33,7 @@ def test_map_float_object_conversion(val):
     assert result == object
 
 
-def test_map_keeps_dtype(na_action):
+def test_map_keeps_dtype(na_action_frame):
     # GH52219
     arr = Series(["a", np.nan, "b"])
     sparse_arr = arr.astype(pd.SparseDtype(object))
@@ -42,15 +42,15 @@ def test_map_keeps_dtype(na_action):
     def func(x):
         return str.upper(x) if not pd.isna(x) else x
 
-    result = df.map(func, na_action=na_action)
+    result = df.map(func, na_action=na_action_frame)
 
     expected_sparse = pd.array(["A", np.nan, "B"], dtype=pd.SparseDtype(object))
     expected_arr = expected_sparse.astype(object)
     expected = DataFrame({"a": expected_arr, "b": expected_sparse})
 
     tm.assert_frame_equal(result, expected)
 
-    result_empty = df.iloc[:0, :].map(func, na_action=na_action)
+    result_empty = df.iloc[:0, :].map(func, na_action=na_action_frame)
     expected_empty = expected.iloc[:0, :]
     tm.assert_frame_equal(result_empty, expected_empty)