period_range and Period should accept date, datetime and pd.Timestamp

pandas-stubs/_libs/tslibs/period.pyi

@@ -5,6 +5,7 @@ from typing import (
 )
 
 import numpy as np
+import pandas
 from pandas import (
     Index,
     PeriodIndex,
@@ -61,7 +62,12 @@ class PeriodMixin:
 class Period(PeriodMixin):
     def __init__(
         self,
-        value: Period | str | None = ...,
+        value: Period
+        | str
+        | datetime.datetime
+        | datetime.date
+        | Timestamp
+        | None = ...,
         freq: str | BaseOffset | None = ...,
         ordinal: int | None = ...,
         year: int | None = ...,

pandas-stubs/core/indexes/period.pyi

@@ -1,4 +1,5 @@
 from collections.abc import Hashable
+import datetime
 from typing import overload
 
 import numpy as np
@@ -84,8 +85,18 @@ class PeriodIndex(  # type: ignore[misc]
     def freqstr(self) -> str: ...
 
 def period_range(
-    start: str | pd.Period | None = ...,
-    end: str | pd.Period | None = ...,
+    start: str
+    | datetime.datetime
+    | datetime.date
+    | pd.Timestamp
+    | pd.Period
+    | None = ...,
+    end: str
+    | datetime.datetime
+    | datetime.date
+    | pd.Timestamp
+    | pd.Period
+    | None = ...,
     periods: int | None = ...,
     freq: str | BaseOffset | None = ...,
     name: Hashable | None = ...,

pandas-stubs/core/series.pyi

@@ -851,7 +851,7 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
     @overload
     def apply(
         self,
-        func: Callable[..., Scalar | Sequence | set | Mapping | None],
+        func: Callable[..., Scalar | Sequence | set | Mapping | NAType | None],
         convertDType: _bool = ...,
         args: tuple = ...,
         **kwds,

tests/test_scalars.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import datetime
 import datetime as dt
 from typing import (
     TYPE_CHECKING,
@@ -1690,6 +1691,20 @@ def test_period_construction() -> None:
         pd.Period,
     )
     check(assert_type(pd.Period(freq="Q", year=2012, quarter=2), pd.Period), pd.Period)
+    check(
+        assert_type(
+            pd.Period(value=datetime.datetime(2012, 1, 1), freq="D"), pd.Period
+        ),
+        pd.Period,
+    )
+    check(
+        assert_type(pd.Period(value=datetime.date(2012, 1, 1), freq="D"), pd.Period),
+        pd.Period,
+    )
+    check(
+        assert_type(pd.Period(value=pd.Timestamp(2012, 1, 1), freq="D"), pd.Period),
+        pd.Period,
+    )
 
 
 def test_period_properties() -> None:

tests/test_series.py

@@ -35,6 +35,7 @@
 )
 import xarray as xr
 
+from pandas._libs.missing import NAType
 from pandas._libs.tslibs.timedeltas import Timedelta
 from pandas._libs.tslibs.timestamps import Timestamp
 from pandas._typing import (
@@ -470,6 +471,8 @@ def get_depth(url: str) -> int:
     ss = s.astype(str)
     check(assert_type(ss.apply(get_depth), pd.Series), pd.Series, np.int64)
 
+    check(assert_type(s.apply(lambda x: pd.NA), pd.Series), pd.Series, NAType)
+
 
 def test_types_element_wise_arithmetic() -> None:
     s = pd.Series([0, 1, -10])

tests/test_timefuncs.py

@@ -838,6 +838,27 @@ def test_period_range() -> None:
         ),
         pd.PeriodIndex,
     )
+    check(
+        assert_type(
+            pd.period_range(pd.Timestamp("2001-01-01"), end="2002-01-01", freq="Q"),
+            pd.PeriodIndex,
+        ),
+        pd.PeriodIndex,
+    )
+    check(
+        assert_type(
+            pd.period_range(dt.datetime(2001, 1, 1), end="2002-01-01", freq="Q"),
+            pd.PeriodIndex,
+        ),
+        pd.PeriodIndex,
+    )
+    check(
+        assert_type(
+            pd.period_range(dt.date(2001, 1, 1), end="2002-01-01", freq="Q"),
+            pd.PeriodIndex,
+        ),
+        pd.PeriodIndex,
+    )
 
 
 def test_to_datetime_scalar() -> None: