numpy_extension_array

Author: cmp0xff

pandas-stubs/core/construction.pyi

@@ -151,7 +151,12 @@ def array(
 ) -> Categorical: ...
 @overload
 def array(
-    data: SequenceNotStr[object] | np.typing.NDArray[np.object_] | RangeIndex,
+    data: (
+        SequenceNotStr[object]
+        | np.typing.NDArray[np.object_]
+        | NumpyExtensionArray
+        | RangeIndex
+    ),
     dtype: str | np.dtype | ExtensionDtype | None = None,
     copy: bool = True,
 ) -> NumpyExtensionArray: ...

tests/arrays/test_numpy_extension_array.py

@@ -0,0 +1,33 @@
+import numpy as np
+import pandas as pd
+from pandas.core.arrays.numpy_ import NumpyExtensionArray
+from typing_extensions import assert_type
+
+from tests import check
+
+
+def test_constructor() -> None:
+    check(
+        assert_type(pd.array([pd.NA, None]), NumpyExtensionArray), NumpyExtensionArray
+    )
+
+    check(
+        assert_type(  # type: ignore[assert-type] # I do not understand
+            pd.array([1, "🐼"]), NumpyExtensionArray
+        ),
+        NumpyExtensionArray,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type] # I do not understand, mypy must have problem with two Generic Variables somehow
+            pd.array(np.array([1, "🐼"], np.object_)), NumpyExtensionArray
+        ),
+        NumpyExtensionArray,
+    )
+    check(
+        assert_type(pd.array(pd.array([pd.NA, None])), NumpyExtensionArray),
+        NumpyExtensionArray,
+    )
+    check(
+        assert_type(pd.array(pd.RangeIndex(0, 1)), NumpyExtensionArray),
+        NumpyExtensionArray,
+    )