tests

Author: cmp0xff

pandas-stubs/core/construction.pyi

@@ -1,6 +1,7 @@
 from collections.abc import Sequence
 from typing import (
     Any,
+    Never,
     overload,
 )
 
@@ -23,6 +24,11 @@ from pandas._libs.tslibs.nattype import NaTType
 from pandas._libs.tslibs.timedeltas import Timedelta
 from pandas._libs.tslibs.timestamps import Timestamp
 from pandas._typing import (
+    BooleanDtypeArg,
+    FloatDtypeArg,
+    IntDtypeArg,
+    TimedeltaDtypeArg,
+    TimestampDtypeArg,
     np_ndarray,
     np_ndarray_anyint,
     np_ndarray_bool,
@@ -35,51 +41,34 @@ from pandas.core.dtypes.dtypes import ExtensionDtype
 
 @overload
 def array(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
-    data: Sequence[bool | np.bool | NAType | None] | np_ndarray_bool | BooleanArray,
-    dtype: PandasBooleanDtypeArg | None = None,
+    data: Sequence[Never] | Index[Never] | Series[Never],
+    dtype: str | np.dtype | ExtensionDtype | None = None,
     copy: bool = True,
-) -> BooleanArray: ...
+) -> ExtensionArray: ...
 @overload
-def array(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
+def array(  # type: ignore[overload-overlap]
     data: Sequence[NAType | None],
     dtype: str | np.dtype | ExtensionDtype | None = None,
     copy: bool = True,
 ) -> NumpyExtensionArray: ...
 @overload
-def array(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
-    data: (
-        Sequence[bool | np.bool | NAType | None]
-        | np_ndarray_bool
-        | BooleanArray
-        | Index[bool]
-        | Series[int]
-    ),
-    dtype: str | np.dtype | ExtensionDtype | None = None,
+def array(  # type: ignore[overload-overlap]
+    data: Sequence[bool | np.bool | NAType | None] | np_ndarray_bool | BooleanArray,
+    dtype: BooleanDtypeArg | None = None,
     copy: bool = True,
 ) -> BooleanArray: ...
 @overload
 def array(  # type: ignore[overload-overlap]
-    data: (
-        Sequence[int | np.integer | NAType | None]
-        | np_ndarray_anyint
-        | IntegerArray
-        | Index[int]
-        | RangeIndex
-        | Series[int]
-    ),
-    dtype: str | np.dtype | ExtensionDtype | None = None,
+    data: Sequence[int | np.integer | NAType | None] | np_ndarray_anyint | IntegerArray,
+    dtype: IntDtypeArg | None = None,
     copy: bool = True,
 ) -> IntegerArray: ...
 @overload
 def array(  # type: ignore[overload-overlap]
     data: (
-        Sequence[float | np.floating | NAType | None]
-        | np_ndarray_float
-        | FloatingArray
-        | Index[float]
-        | Series[float]
+        Sequence[float | np.floating | NAType | None] | np_ndarray_float | FloatingArray
     ),
-    dtype: str | np.dtype | ExtensionDtype | None = None,
+    dtype: FloatDtypeArg | None = None,
     copy: bool = True,
 ) -> FloatingArray: ...
 @overload
@@ -91,7 +80,7 @@ def array(  # type: ignore[overload-overlap]
         | DatetimeIndex
         | Series[Timestamp]
     ),
-    dtype: str | np.dtype | ExtensionDtype | None = None,
+    dtype: TimestampDtypeArg | None = None,
     copy: bool = True,
 ) -> DatetimeArray: ...
 @overload
@@ -103,12 +92,18 @@ def array(  # type: ignore[overload-overlap]
         | TimedeltaIndex
         | Series[Timedelta]
     ),
-    dtype: str | np.dtype | ExtensionDtype | None = None,
+    dtype: TimedeltaDtypeArg | None = None,
     copy: bool = True,
 ) -> TimedeltaArray: ...
 @overload
 def array(
-    data: Sequence[Any] | np_ndarray | ExtensionArray | Index | Series,
+    data: Sequence[object] | np.typing.NDArray[np.object_] | RangeIndex,
     dtype: str | np.dtype | ExtensionDtype | None = None,
     copy: bool = True,
 ) -> NumpyExtensionArray: ...
+@overload
+def array(
+    data: Sequence[Any] | np_ndarray | ExtensionArray | Index[Any] | Series[Any],
+    dtype: str | np.dtype | ExtensionDtype | None = None,
+    copy: bool = True,
+) -> ExtensionArray: ...

tests/arrays/test_boolean_arrays.py

@@ -11,3 +11,7 @@ def test_construction() -> None:
     check(assert_type(pd.array([True, np.bool(True)]), BooleanArray), BooleanArray)
     check(assert_type(pd.array([True, None]), BooleanArray), BooleanArray)
     check(assert_type(pd.array([True, pd.NA]), BooleanArray), BooleanArray)
+
+    check(assert_type(pd.array(np.array([1], np.bool_)), BooleanArray), BooleanArray)
+
+    check(assert_type(pd.array(pd.array([True])), BooleanArray), BooleanArray)

tests/arrays/test_integer_arrays.py

@@ -11,3 +11,12 @@ def test_construction() -> None:
     check(assert_type(pd.array([1, np.int64(1)]), IntegerArray), IntegerArray)
     check(assert_type(pd.array([1, None]), IntegerArray), IntegerArray)
     check(assert_type(pd.array([1, pd.NA, None]), IntegerArray), IntegerArray)
+
+    check(
+        assert_type(  # type: ignore[assert-type]  # I do not understand
+            pd.array(np.array([1], np.int64)), IntegerArray
+        ),
+        IntegerArray,
+    )
+
+    check(assert_type(pd.array(pd.array([1])), IntegerArray), IntegerArray)