more tests

Author: cmp0xff

pandas-stubs/core/construction.pyi

@@ -1,7 +1,10 @@
 from collections.abc import Sequence
+from datetime import (
+    datetime,
+    timedelta,
+)
 from typing import (
     Any,
-    Never,
     overload,
 )
 
@@ -27,6 +30,7 @@ from pandas.core.indexes.period import PeriodIndex
 from pandas.core.indexes.range import RangeIndex
 from pandas.core.indexes.timedeltas import TimedeltaIndex
 from pandas.core.series import Series
+from typing_extensions import Never
 
 from pandas._libs.interval import Interval
 from pandas._libs.missing import NAType
@@ -95,7 +99,8 @@ def array(  # type: ignore[overload-overlap]
 @overload
 def array(  # type: ignore[overload-overlap]
     data: (
-        Sequence[Timestamp | np.datetime64 | NaTType | None]
+        Sequence[datetime | NaTType | None]
+        | Sequence[np.datetime64 | NaTType | None]
         | np_ndarray_dt
         | DatetimeArray
         | DatetimeIndex
@@ -107,7 +112,7 @@ def array(  # type: ignore[overload-overlap]
 @overload
 def array(  # type: ignore[overload-overlap]
     data: (
-        Sequence[Timedelta | np.timedelta64 | NaTType | None]
+        Sequence[timedelta | np.timedelta64 | NaTType | None]
         | np_ndarray_td
         | TimedeltaArray
         | TimedeltaIndex

tests/arrays/test_boolean_arrays.py renamed to tests/arrays/test_boolean_array.py

@@ -6,7 +6,7 @@
 from tests import check
 
 
-def test_construction() -> None:
+def test_constructor() -> None:
     check(assert_type(pd.array([True]), BooleanArray), BooleanArray)
     check(assert_type(pd.array([True, np.bool(True)]), BooleanArray), BooleanArray)
     check(assert_type(pd.array([True, None]), BooleanArray), BooleanArray)

tests/arrays/test_datetime_array.py

@@ -0,0 +1,34 @@
+from datetime import datetime
+
+import numpy as np
+import pandas as pd
+from pandas.core.arrays.datetimes import DatetimeArray
+from typing_extensions import assert_type
+
+from tests import check
+
+
+def test_constructor() -> None:
+    dt = datetime(2025, 11, 10)
+    check(assert_type(pd.array([dt]), DatetimeArray), DatetimeArray)
+    check(assert_type(pd.array([dt, pd.Timestamp(dt)]), DatetimeArray), DatetimeArray)
+    check(assert_type(pd.array([dt, None]), DatetimeArray), DatetimeArray)
+    check(assert_type(pd.array([dt, pd.NaT, None]), DatetimeArray), DatetimeArray)
+
+    np_dt = np.datetime64(dt)
+    check(assert_type(pd.array([np_dt]), DatetimeArray), DatetimeArray)
+    check(assert_type(pd.array([np_dt, None]), DatetimeArray), DatetimeArray)
+    check(assert_type(pd.array([np_dt, pd.NaT]), DatetimeArray), DatetimeArray)
+
+    check(
+        assert_type(  # type: ignore[assert-type]  # I do not understand
+            pd.array(np.array([dt], np.datetime64)), DatetimeArray
+        ),
+        DatetimeArray,
+    )
+
+    check(assert_type(pd.array(pd.array([dt])), DatetimeArray), DatetimeArray)
+
+    check(assert_type(pd.array(pd.Index([dt])), DatetimeArray), DatetimeArray)
+
+    check(assert_type(pd.array(pd.Series([dt])), DatetimeArray), DatetimeArray)

tests/arrays/test_floating_array.py

@@ -0,0 +1,22 @@
+import numpy as np
+import pandas as pd
+from pandas.core.arrays.floating import FloatingArray
+from typing_extensions import assert_type
+
+from tests import check
+
+
+def test_constructor() -> None:
+    check(assert_type(pd.array([1.0]), FloatingArray), FloatingArray)
+    check(assert_type(pd.array([1.0, np.float64(1)]), FloatingArray), FloatingArray)
+    check(assert_type(pd.array([1.0, None]), FloatingArray), FloatingArray)
+    check(assert_type(pd.array([1.0, pd.NA, None]), FloatingArray), FloatingArray)
+
+    check(
+        assert_type(  # type: ignore[assert-type]  # I do not understand
+            pd.array(np.array([1.0], np.float64)), FloatingArray
+        ),
+        FloatingArray,
+    )
+
+    check(assert_type(pd.array(pd.array([1.0])), FloatingArray), FloatingArray)

tests/arrays/test_integer_arrays.py renamed to tests/arrays/test_integer_array.py

@@ -6,7 +6,7 @@
 from tests import check
 
 
-def test_construction() -> None:
+def test_constructor() -> None:
     check(assert_type(pd.array([1]), IntegerArray), IntegerArray)
     check(assert_type(pd.array([1, np.int64(1)]), IntegerArray), IntegerArray)
     check(assert_type(pd.array([1, None]), IntegerArray), IntegerArray)