Improved the type stubs in the _libs directory to help with type checking.

Author: msfterictraut

pandas/_libs/interval.pyi

@@ -0,0 +1,47 @@
+from typing import FrozenSet, Tuple
+
+import numpy as np
+
+class IntervalMixin:
+    @property
+    def _left(self) -> object: ...
+    @property
+    def _right(self) -> object: ...
+    @property
+    def closed_left(self) -> bool: ...
+    @property
+    def closed_right(self) -> bool: ...
+    @property
+    def open_left(self) -> bool: ...
+    @property
+    def open_right(self) -> bool: ...
+    @property
+    def mid(self) -> float: ...
+    @property
+    def length(self) -> float: ...
+    @property
+    def is_empty(self) -> bool: ...
+    def _check_closed_matches(self, other: IntervalMixin, name: str=...) -> None: ...
+
+class Interval(IntervalMixin):
+    def __init__(self, left: object, right: object, closed: str=...) -> None: ...
+    @property
+    def closed(self) -> str: ...
+    @property
+    def left(self) -> object: ...
+    @property
+    def right(self) -> object: ...
+    def __str__(self) -> str: ...
+    def __add__(self, y: Interval) -> Interval: ...
+    def __sub__(self, y: Interval) -> Interval: ...
+    def __mul__(self, y: Interval) -> Interval: ...
+    def __trudeiv__(self, y: Interval) -> Interval: ...
+    def __floordiv__(self, y: Interval) -> Interval: ...
+    def __overlaps__(self, other: Interval) -> bool: ...
+
+VALID_CLOSED: FrozenSet[str]
+
+def intervals_to_interval_bounds(
+    intervals: np.ndarray,
+    valdiate_closed: bool = ...
+) -> Tuple[object, object, str]: ...

pandas/_libs/tslibs/ccalendar.pyi

@@ -8,5 +8,7 @@ def get_firstbday(year: int, month: int) -> int: ...
 def get_lastbday(year: int, month: int) -> int: ...
 def get_day_of_year(year: int, month: int, day: int) -> int: ...
 def get_iso_calendar(year: int, month: int, day: int) -> tuple[int, int, int]: ...
+def is_leapyear(year: int) -> bool: ...
 def get_week_of_year(year: int, month: int, day: int) -> int: ...
 def get_days_in_month(year: int, month: int) -> int: ...
+def dayofweek(y: int, m: int, d: int) -> int: ...

pandas/_libs/tslibs/nattype.pyi

@@ -14,6 +14,7 @@ iNaT: int
 nat_strings: set[str]
 
 def is_null_datetimelike(val: object, inat_is_null: bool = ...) -> bool: ...
+def checknull_with_nat(val: object) -> bool: ...
 
 class NaTType(datetime):
     value: np.int64

pandas/_libs/tslibs/offsets.pyi

@@ -0,0 +1,102 @@
+from __future__ import annotations
+from datetime import datetime
+from typing import Any, Tuple, Union
+from datetime import timedelta
+
+class BaseOffset:
+    def __init__(self, n: int = ..., normalize: bool = ...) -> None: ...
+    def __eq__(self, other) -> bool: ...
+    def __ne__(self, other) -> bool: ...
+    def __hash__(self) -> int: ...
+    @property
+    def kwds(self) -> dict: ...
+    @property
+    def base(self) -> BaseOffset: ...
+    def __add__(self, other) -> BaseOffset: ...
+    def __sub__(self, other) -> BaseOffset: ...
+    def __call__(self, other): ...
+    def __mul__(self, other): ...
+    def __neg__(self) -> BaseOffset: ...
+    def copy(self) -> BaseOffset: ...
+    def __repr__(self) -> str: ...
+    @property
+    def name(self) -> str: ...
+    @property
+    def rule_code(self) -> str: ...
+    def freqstr(self) -> str: ...
+    # Next one is problematic due to circular imports
+    #def apply_index(self, dtindex: DatetimeIndex) -> DatetimeIndex: ...
+    def apply_index(self, dtindex): ...
+    def _apply_array(self, dtarr) -> None: ...
+    def rollback(self, dt: datetime) -> datetime: ...
+    def rollforward(self, dt: datetime) -> datetime: ...
+    def is_on_offset(self, dt: datetime) -> bool: ...
+    def __setstate__(self, state) -> None: ...
+    def __getstate__(self): ...
+    @property
+    def nanos(self) -> int: ...
+    def onOffset(self, dt: datetime) -> bool: ...
+    def isAnchored(self) -> bool: ...
+    def is_anchored(self) -> bool: ...
+
+class SingleConstructorOffset(BaseOffset):
+    @classmethod
+    def _from_name(cls, suffix=None): ...
+    def __reduce__(self): ...
+
+def to_offset(freq: Union[str, Tuple, timedelta, BaseOffset, None]) -> Union[BaseOffset, None]: ...
+
+class Tick(SingleConstructorOffset):
+    def __init__(self, n: int = ..., normalize: bool = ...) -> None: ...
+
+class Day(Tick): ...
+class Hour(Tick): ...
+class Minute(Tick): ...
+class Second(Tick): ...
+class Milli(Tick): ...
+class Micro(Tick): ...
+class Nano(Tick): ...
+class RelativeDeltaOffset(BaseOffset):
+    def __init__(self, n: int = ..., normalize: bool = ..., **kwds: Any) -> None: ...
+class BusinessMixin(SingleConstructorOffset): ...
+class BusinessDay(BusinessMixin): ...
+class BusinessHour(BusinessMixin): ...
+class WeekOfMonthMixin(SingleConstructorOffset): ...
+class YearOffset(SingleConstructorOffset): ...
+class BYearEnd(YearOffset): ...
+class BYearBegin(YearOffset): ...
+class YearEnd(YearOffset): ...
+class YearBegin(YearOffset): ...
+class QuarterOffset(SingleConstructorOffset): ...
+class BQuarterEnd(QuarterOffset): ...
+class BQuarterBegin(QuarterOffset): ...
+class QuarterEnd(QuarterOffset): ...
+class QuarterBegin(QuarterOffset): ...
+class MonthOffset(SingleConstructorOffset): ...
+class MonthEnd(MonthOffset): ...
+class MonthBegin(MonthOffset): ...
+class BusinessMonthEnd(MonthOffset): ...
+class BusinessMonthBegin(MonthOffset): ...
+class SemiMonthOffset(SingleConstructorOffset): ...
+class SemiMonthEnd(SemiMonthOffset): ...
+class SemiMonthBegin(SemiMonthOffset): ...
+class Week(SingleConstructorOffset): ...
+class WeekOfMonth(WeekOfMonthMixin): ...
+class LastWeekOfMonth(WeekOfMonthMixin): ...
+class FY5253Mixin(SingleConstructorOffset): ...
+class FY5253(FY5253Mixin): ...
+class FY5253Quarter(FY5253Mixin): ...
+class Easter(SingleConstructorOffset): ...
+class _CustomBusinessMonth(BusinessMixin): ...
+class CustomBusinessDay(BusinessDay): ...
+class CustomBusinessHour(BusinessHour): ...
+class CustomBusinessMonthEnd(_CustomBusinessMonth): ...
+class CustomBusinessMonthBegin(_CustomBusinessMonth): ...
+class DateOffset(RelativeDeltaOffset): ...
+
+BDay = BusinessDay
+BMonthEnd = BusinessMonthEnd
+BMonthBegin = BusinessMonthBegin
+CBMonthEnd = CustomBusinessMonthEnd
+CBMonthBegin = CustomBusinessMonthBegin
+CDay = CustomBusinessDay

pandas/_libs/tslibs/period.pyi

@@ -1,3 +1,4 @@
+from datetime import datetime
 from typing import Literal
 
 import numpy as np
@@ -14,6 +15,9 @@ from pandas._typing import (
 INVALID_FREQ_ERR_MSG: str
 DIFFERENT_FREQ: str
 
+def is_period_object(obj: object) -> bool: ...
+def get_period_ordinal(dts: datetime, freq: int) -> int: ...
+
 class IncompatibleFrequency(ValueError): ...
 
 def periodarr_to_dt64arr(