two more test files

Author: cmp0xff

pandas-stubs/core/arrays/__init__.pyi

@@ -12,10 +12,7 @@ from pandas.core.arrays.integer import IntegerArray
 from pandas.core.arrays.interval import IntervalArray
 from pandas.core.arrays.masked import BaseMaskedArray
 from pandas.core.arrays.numpy_ import NumpyExtensionArray
-from pandas.core.arrays.period import (
-    PeriodArray,
-    period_array,
-)
+from pandas.core.arrays.period import PeriodArray
 from pandas.core.arrays.sparse import SparseArray
 from pandas.core.arrays.string_ import StringArray
 from pandas.core.arrays.string_arrow import ArrowStringArray
@@ -39,5 +36,4 @@ __all__ = [
     "SparseArray",
     "StringArray",
     "TimedeltaArray",
-    "period_array",
 ]

pandas-stubs/core/arrays/period.pyi

@@ -1,19 +1,12 @@
-from collections.abc import Sequence
-
 from pandas import PeriodDtype
 from pandas.core.arrays.datetimelike import (
     DatelikeOps,
     DatetimeLikeArrayMixin,
 )
 
 from pandas._libs.tslibs import Timestamp
-from pandas._libs.tslibs.offsets import (
-    BaseOffset,
-    Tick,
-)
 from pandas._libs.tslibs.period import Period
 from pandas._typing import (
-    AnyArrayLike,
     NpDtype,
     PeriodFrequency,
     np_1darray,
@@ -54,9 +47,3 @@ class PeriodArray(DatetimeLikeArrayMixin, DatelikeOps):
         self, freq: PeriodFrequency | None = None, how: str = ...
     ) -> Timestamp: ...
     def asfreq(self, freq: str | None = ..., how: str = "E") -> Period: ...
-
-def period_array(
-    data: Sequence[Period | str | None] | AnyArrayLike,
-    freq: str | Tick | BaseOffset | None = None,
-    copy: bool = False,
-) -> PeriodArray: ...

pandas-stubs/core/construction.pyi

@@ -128,11 +128,13 @@ def array(
     copy: bool = True,
 ) -> StringArray: ...
 @overload
-def array(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
-    data: Sequence[Period | NAType | None] | PeriodArray | PeriodIndex | Series[Period],
+def array(  # type: ignore[overload-overlap]
+    data: (
+        Sequence[Period | NaTType | None] | PeriodArray | PeriodIndex | Series[Period]
+    ),
     dtype: PeriodDtype | None = None,
     copy: bool = True,
-) -> PeriodIndex: ...
+) -> PeriodArray: ...
 @overload
 def array(  # type: ignore[overload-overlap]
     data: (
@@ -157,11 +159,11 @@ def array(
     copy: bool = True,
 ) -> NumpyExtensionArray: ...
 @overload
-def array(  # type: ignore[overload-overlap]
+def array(
     data: SparseArray | SparseIndex,
     dtype: str | np.dtype | ExtensionDtype | None = None,
     copy: bool = True,
-) -> SparseIndex: ...
+) -> SparseArray: ...
 @overload
 def array(
     data: ArrowExtensionArray,

tests/arrays/test_period_array.py

@@ -0,0 +1,38 @@
+import pandas as pd
+from pandas.core.arrays.period import PeriodArray
+from typing_extensions import assert_type
+
+from tests import check
+
+
+def test_constructor() -> None:
+    prd = pd.Period("2023-01-01")
+    check(
+        assert_type(  # type: ignore[assert-type]  # I do not understand
+            pd.array([prd]), PeriodArray
+        ),
+        PeriodArray,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]  # I do not understand
+            pd.array([prd, None]), PeriodArray
+        ),
+        PeriodArray,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]  # I do not understand
+            pd.array([prd, pd.NaT, None]), PeriodArray
+        ),
+        PeriodArray,
+    )
+
+    check(
+        assert_type(  # type: ignore[assert-type]  # I do not understand
+            pd.array(pd.array([prd])), PeriodArray
+        ),
+        PeriodArray,
+    )
+
+    check(assert_type(pd.array(pd.Index([prd])), PeriodArray), PeriodArray)
+
+    check(assert_type(pd.array(pd.Series([prd])), PeriodArray), PeriodArray)

tests/arrays/test_string_array.py

@@ -0,0 +1,18 @@
+import numpy as np
+import pandas as pd
+from pandas.core.arrays.string_ import StringArray
+from typing_extensions import assert_type
+
+from tests import check
+
+
+def test_constructor() -> None:
+    check(assert_type(pd.array(["🐼"]), StringArray), StringArray)
+    check(
+        assert_type(pd.array(["🐼", np.str_("🐼")]), StringArray),
+        StringArray,
+    )
+    check(assert_type(pd.array(["🐼", None]), StringArray), StringArray)
+    check(assert_type(pd.array(["🐼", pd.NA, None]), StringArray), StringArray)
+
+    check(assert_type(pd.array(pd.array(["🐼"])), StringArray), StringArray)