updated test and map condition

Author: arthurlw

pandas/core/arrays/arrow/array.py

@@ -1481,12 +1481,8 @@ def to_numpy(
         return result
 
     def map(self, mapper, na_action: Literal["ignore"] | None = None):
-        if is_numeric_dtype(self.dtype):
+        if is_numeric_dtype(self.dtype) or self.dtype.kind in "mM":
             return map_array(self.to_numpy(), mapper, na_action=na_action)
-        elif self.dtype == "timestamp[ns][pyarrow]":
-            return map_array(
-                self.to_numpy(dtype="datetime64[ns]"), mapper, na_action=na_action
-            )
         else:
             return super().map(mapper, na_action)
 

pandas/tests/extension/test_arrow.py

@@ -281,7 +281,7 @@ def test_compare_scalar(self, data, comparison_op):
     @pytest.mark.parametrize("na_action", [None, "ignore"])
     def test_map(self, data_missing, na_action):
         if data_missing.dtype.kind in "mM":
-            result = data_missing.map(lambda x: x, na_action=na_action)
+            result = data_missing.map(lambda x: x, na_action=na_action).to_numpy(dtype=object)
             expected = data_missing.to_numpy(dtype=object)
             tm.assert_numpy_array_equal(result, expected)
         else: