check ExtensionType in is_datetime64_any_dtype for array-likes (pandas-dev#57060)

* check for ExtensionType in is_datetime64_any_dtype

* use pre-commit

* add test and move doc entry

* check not date in test

* fix condition

* check type is not datetime.date

* fix comparison

* move description to ExtensionArray

* return True for date types

Author: jmoralez

pandas/core/dtypes/common.py

@@ -890,7 +890,11 @@ def is_datetime64_any_dtype(arr_or_dtype) -> bool:
         tipo = _get_dtype(arr_or_dtype)
     except TypeError:
         return False
-    return lib.is_np_dtype(tipo, "M") or isinstance(tipo, DatetimeTZDtype)
+    return (
+        lib.is_np_dtype(tipo, "M")
+        or isinstance(tipo, DatetimeTZDtype)
+        or (isinstance(tipo, ExtensionDtype) and tipo.kind == "M")
+    )
 
 
 def is_datetime64_ns_dtype(arr_or_dtype) -> bool:

pandas/tests/extension/test_arrow.py

@@ -53,6 +53,7 @@
 from pandas.api.extensions import no_default
 from pandas.api.types import (
     is_bool_dtype,
+    is_datetime64_any_dtype,
     is_float_dtype,
     is_integer_dtype,
     is_numeric_dtype,
@@ -1531,6 +1532,14 @@ def test_is_unsigned_integer_dtype(data):
         assert not is_unsigned_integer_dtype(data)
 
 
+def test_is_datetime64_any_dtype(data):
+    pa_type = data.dtype.pyarrow_dtype
+    if pa.types.is_timestamp(pa_type) or pa.types.is_date(pa_type):
+        assert is_datetime64_any_dtype(data)
+    else:
+        assert not is_datetime64_any_dtype(data)
+
+
 def test_is_float_dtype(data):
     pa_type = data.dtype.pyarrow_dtype
     if pa.types.is_floating(pa_type):