fixed some errors

Author: pedromfdiogo

doc/source/whatsnew/v3.0.0.rst

@@ -864,7 +864,6 @@ ExtensionArray
 - Bug in constructing pandas data structures when passing into ``dtype`` a string of the type followed by ``[pyarrow]`` while PyArrow is not installed would raise ``NameError`` rather than ``ImportError`` (:issue:`57928`)
 - Bug in various :class:`DataFrame` reductions for pyarrow temporal dtypes returning incorrect dtype when result was null (:issue:`59234`)
 
-
 Styler
 ^^^^^^
 - Bug in :meth:`Styler.to_latex` where styling column headers when combined with a hidden index or hidden index-levels is fixed.

pandas/tests/arrays/masked/test_basemaskedarray_map.py

@@ -1,18 +1,16 @@
-import pandas as pd
-
+from pandas import Series, NA, isna
+from pandas.testing import assert_series_equal
 
 def test_basemaskedarray_map():
-    for dtype, data, expected_data in [
-        ("Int32", [1, 2, None, 4], [2, 3, pd.NA, 5]),
-    ]:
-        s = pd.Series(data, dtype=dtype)
+    
+        s = Series([1, 2, None, 4], dtype="Int32")
 
         def transform(x):
-            if x is None:
+            if isna(x):
                 return x
             return x + 1
 
         result = s.map(transform)
-        expected = pd.Series(expected_data, dtype=result.dtype)
+        expected = Series([2, 3, NA, 5], dtype=result.dtype)
 
-        assert result.tolist() == expected.tolist()
+        assert_series_equal(result, expected)

pandas/tests/extension/test_masked.py

@@ -191,7 +191,7 @@ def test_map_na_action_ignore(self, data_missing_for_sorting):
         if data_missing_for_sorting.dtype.kind == "b":
             expected = np.array([False, pd.NA, False], dtype=object)
         else:
-            expected = np.array([zero, np.nan, zero])
+            expected = np.array([zero, pd.NA, zero])
             for i in range(len(result)):
                 if result[i] is pd.NA:
                     result[i] = "nan"