dtype numpy extension array

Author: cmp0xff

pandas-stubs/core/construction.pyi

@@ -40,12 +40,12 @@ from pandas._libs.tslibs.period import Period
 from pandas._libs.tslibs.timedeltas import Timedelta
 from pandas._libs.tslibs.timestamps import Timestamp
 from pandas._typing import (
-    BuiltinFloatDtypeArg,
-    BuiltinIntDtypeArg,
-    BuiltinStrDtypeArg,
+    BuiltinDtypeArg,
     CategoryDtypeArg,
     IntervalT,
-    NumpyDtypeArg,
+    NumpyNotTimeDtypeArg,
+    NumpyTimedeltaDtypeArg,
+    NumpyTimestampDtypeArg,
     PandasBooleanDtypeArg,
     PandasFloatDtypeArg,
     PandasIntDtypeArg,
@@ -119,15 +119,15 @@ def array(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlapping
 @overload
 def array(  # type: ignore[overload-overlap]
     data: Sequence[int | np.integer | NAType | None] | np_ndarray_anyint | IntegerArray,
-    dtype: BuiltinIntDtypeArg | PandasIntDtypeArg | PandasUIntDtypeArg | None = None,
+    dtype: PandasIntDtypeArg | PandasUIntDtypeArg | None = None,
     copy: bool = True,
 ) -> IntegerArray: ...
 @overload
 def array(  # type: ignore[overload-overlap]
     data: (
         Sequence[float | np.floating | NAType | None] | np_ndarray_float | FloatingArray
     ),
-    dtype: BuiltinFloatDtypeArg | PandasFloatDtypeArg | None = None,
+    dtype: PandasFloatDtypeArg | None = None,
     copy: bool = True,
 ) -> FloatingArray: ...
 @overload
@@ -140,7 +140,7 @@ def array(  # type: ignore[overload-overlap]
         | DatetimeIndex
         | Series[Timestamp]
     ),
-    dtype: PandasTimestampDtypeArg | None = None,
+    dtype: PandasTimestampDtypeArg | NumpyTimestampDtypeArg | None = None,
     copy: bool = True,
 ) -> DatetimeArray: ...
 @overload
@@ -152,13 +152,13 @@ def array(
         | TimedeltaIndex
         | Series[Timedelta]
     ),
-    dtype: None = None,
+    dtype: NumpyTimedeltaDtypeArg | None = None,
     copy: bool = True,
 ) -> TimedeltaArray: ...
 @overload
 def array(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
     data: SequenceNotStr[str | np.str_ | NAType | None] | np_ndarray_str | StringArray,
-    dtype: BuiltinStrDtypeArg | PandasStrDtypeArg | None = None,
+    dtype: PandasStrDtypeArg | None = None,
     copy: bool = True,
 ) -> StringArray: ...
 @overload
@@ -175,7 +175,7 @@ def array(  # type: ignore[overload-overlap]
 @overload
 def array(
     data: SequenceNotStr[object] | np_ndarray | NumpyExtensionArray | RangeIndex,
-    dtype: NumpyDtypeArg | None = None,
+    dtype: BuiltinDtypeArg | NumpyNotTimeDtypeArg | None = None,
     copy: bool = True,
 ) -> NumpyExtensionArray: ...
 @overload

tests/arrays/test_numpy_extension_array.py

@@ -1,9 +1,15 @@
 import numpy as np
 import pandas as pd
 from pandas.core.arrays.numpy_ import NumpyExtensionArray
+import pytest
 from typing_extensions import assert_type
 
-from tests import check
+from tests import (
+    BuiltinDtypeArg,
+    NumpyNotTimeDtypeArg,
+    check,
+    get_dtype,
+)
 
 
 def test_constructor() -> None:
@@ -31,3 +37,17 @@ def test_constructor() -> None:
         assert_type(pd.array(pd.RangeIndex(0, 1)), NumpyExtensionArray),
         NumpyExtensionArray,
     )
+
+
+@pytest.mark.parametrize("dtype", get_dtype(BuiltinDtypeArg | NumpyNotTimeDtypeArg))
+def test_constructors_dtype(dtype: BuiltinDtypeArg | NumpyNotTimeDtypeArg):
+    if dtype == "V" or "void" in str(dtype):
+        check(
+            assert_type(pd.array([b"1"], dtype=dtype), NumpyExtensionArray),
+            NumpyExtensionArray,
+        )
+    else:
+        check(
+            assert_type(pd.array([1], dtype=dtype), NumpyExtensionArray),
+            NumpyExtensionArray,
+        )