feat: truediv, also pandas-dev/pandas#62712

Author: cmp0xff

pandas-stubs/_libs/tslibs/period.pyi

@@ -99,7 +99,7 @@ class Period(PeriodMixin):
     # Ignored due to indecipherable error from mypy:
     # Forward operator "__add__" is not callable  [misc]
     @overload
-    def __radd__(self, other: _PeriodAddSub) -> Self: ...  # type: ignore[misc]
+    def __radd__(self, other: _PeriodAddSub) -> Self: ...
     @overload
     def __radd__(self, other: NaTType) -> NaTType: ...
     # Real signature is -> PeriodIndex, but conflicts with Index.__add__

pandas-stubs/_libs/tslibs/timedeltas.pyi

@@ -321,7 +321,7 @@ class Timedelta(timedelta):
     # for le, lt ge and gt
     # Override due to more types supported than timedelta
     @overload  # type: ignore[override]
-    def __le__(self, other: timedelta | Timedelta | np.timedelta64) -> bool: ...  # type: ignore[misc]
+    def __le__(self, other: timedelta | Timedelta | np.timedelta64) -> bool: ...
     @overload
     def __le__(self, other: TimedeltaIndex) -> np_1darray[np.bool]: ...
     @overload
@@ -332,7 +332,7 @@ class Timedelta(timedelta):
     def __le__(self, other: Series[pd.Timedelta]) -> Series[bool]: ...
     # Override due to more types supported than timedelta
     @overload  # type: ignore[override]
-    def __lt__(self, other: timedelta | Timedelta | np.timedelta64) -> bool: ...  # type: ignore[misc]
+    def __lt__(self, other: timedelta | Timedelta | np.timedelta64) -> bool: ...
     @overload
     def __lt__(self, other: TimedeltaIndex) -> np_1darray[np.bool]: ...
     @overload
@@ -343,7 +343,7 @@ class Timedelta(timedelta):
     def __lt__(self, other: Series[pd.Timedelta]) -> Series[bool]: ...
     # Override due to more types supported than timedelta
     @overload  # type: ignore[override]
-    def __ge__(self, other: timedelta | Timedelta | np.timedelta64) -> bool: ...  # type: ignore[misc]
+    def __ge__(self, other: timedelta | Timedelta | np.timedelta64) -> bool: ...
     @overload
     def __ge__(self, other: TimedeltaIndex) -> np_1darray[np.bool]: ...
     @overload
@@ -354,7 +354,7 @@ class Timedelta(timedelta):
     def __ge__(self, other: Series[pd.Timedelta]) -> Series[bool]: ...
     # Override due to more types supported than timedelta
     @overload  # type: ignore[override]
-    def __gt__(self, other: timedelta | Timedelta | np.timedelta64) -> bool: ...  # type: ignore[misc]
+    def __gt__(self, other: timedelta | Timedelta | np.timedelta64) -> bool: ...
     @overload
     def __gt__(self, other: TimedeltaIndex) -> np_1darray[np.bool]: ...
     @overload

pandas-stubs/_libs/tslibs/timestamps.pyi

@@ -175,7 +175,7 @@ class Timestamp(datetime, SupportsIndex):
     # Mypy complains Forward operator "<inequality op>" is not callable, so ignore misc
     # for le, lt ge and gt
     @overload  # type: ignore[override]
-    def __le__(self, other: Timestamp | datetime | np.datetime64) -> bool: ...  # type: ignore[misc]
+    def __le__(self, other: Timestamp | datetime | np.datetime64) -> bool: ...
     @overload
     def __le__(self, other: DatetimeIndex) -> np_1darray[np.bool]: ...
     @overload
@@ -185,7 +185,7 @@ class Timestamp(datetime, SupportsIndex):
     @overload
     def __le__(self, other: Series[Timestamp]) -> Series[bool]: ...
     @overload  # type: ignore[override]
-    def __lt__(self, other: Timestamp | datetime | np.datetime64) -> bool: ...  # type: ignore[misc]
+    def __lt__(self, other: Timestamp | datetime | np.datetime64) -> bool: ...
     @overload
     def __lt__(self, other: DatetimeIndex) -> np_1darray[np.bool]: ...
     @overload
@@ -195,7 +195,7 @@ class Timestamp(datetime, SupportsIndex):
     @overload
     def __lt__(self, other: Series[Timestamp]) -> Series[bool]: ...
     @overload  # type: ignore[override]
-    def __ge__(self, other: Timestamp | datetime | np.datetime64) -> bool: ...  # type: ignore[misc]
+    def __ge__(self, other: Timestamp | datetime | np.datetime64) -> bool: ...
     @overload
     def __ge__(self, other: DatetimeIndex) -> np_1darray[np.bool]: ...
     @overload
@@ -205,7 +205,7 @@ class Timestamp(datetime, SupportsIndex):
     @overload
     def __ge__(self, other: Series[Timestamp]) -> Series[bool]: ...
     @overload  # type: ignore[override]
-    def __gt__(self, other: Timestamp | datetime | np.datetime64) -> bool: ...  # type: ignore[misc]
+    def __gt__(self, other: Timestamp | datetime | np.datetime64) -> bool: ...
     @overload
     def __gt__(self, other: DatetimeIndex) -> np_1darray[np.bool]: ...
     @overload

tests/indexes/arithmetic/float/test_truediv.py

@@ -0,0 +1,103 @@
+import numpy as np
+from numpy import typing as npt  # noqa: F401
+import pandas as pd
+import pytest
+from typing_extensions import assert_type
+
+from tests import (
+    TYPE_CHECKING_INVALID_USAGE,
+    check,
+)
+
+
+@pytest.fixture
+def left() -> "pd.Index[float]":
+    """Left operand"""
+    lo = pd.Index([1.0, 2.0, 3.0])
+    return check(assert_type(lo, "pd.Index[float]"), pd.Index, np.floating)
+
+
+def test_truediv_py_scalar(left: "pd.Index[float]") -> None:
+    """Test pd.Index[float] / Python native scalars"""
+    b, i, f, c = True, 1, 1.0, 1j
+
+    check(assert_type(left / b, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / i, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / f, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+    check(assert_type(b / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(i / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(f / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(c / left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+
+def test_truediv_py_sequence(left: "pd.Index[float]") -> None:
+    """Test pd.Index[float] / Python native sequences"""
+    b, i, f, c = [True, False, True], [2, 3, 5], [1.0, 2.0, 3.0], [1j, 1j, 4j]
+
+    check(assert_type(left / b, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / i, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / f, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+    check(assert_type(b / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(i / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(f / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(c / left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+
+def test_truediv_numpy_array(left: "pd.Index[float]") -> None:
+    """Test pd.Index[float] / numpy arrays"""
+    b = np.array([True, False, True], np.bool_)
+    i = np.array([2, 3, 5], np.int64)
+    f = np.array([1.0, 2.0, 3.0], np.float64)
+    c = np.array([1.1j, 2.2j, 4.1j], np.complex128)
+
+    check(assert_type(left / b, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / i, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / f, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+    # `numpy` typing gives the corresponding `ndarray`s in the static type
+    # checking, where our `__rtruediv__` cannot override. At runtime, they return
+    # `Series` with the correct element type.
+    check(assert_type(b / left, "npt.NDArray[np.float64]"), pd.Index, np.floating)
+    check(assert_type(i / left, "npt.NDArray[np.float64]"), pd.Index, np.floating)
+    check(assert_type(f / left, "npt.NDArray[np.float64]"), pd.Index, np.floating)
+    check(
+        assert_type(c / left, "npt.NDArray[np.complex128]"),
+        pd.Index,
+        np.complexfloating,
+    )
+
+
+def test_truediv_pd_scalar(left: "pd.Index[float]") -> None:
+    """Test pd.Index[float] / pandas scalars"""
+    s, d = pd.Timestamp(2025, 9, 24), pd.Timedelta(seconds=1)
+
+    if TYPE_CHECKING_INVALID_USAGE:
+        _00 = left / s  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+        _01 = left / d  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+
+    if TYPE_CHECKING_INVALID_USAGE:
+        _10 = s / left  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+    check(assert_type(d / left, "pd.Index[pd.Timedelta]"), pd.Index, pd.Timedelta)
+
+
+def test_truediv_pd_index(left: "pd.Index[float]") -> None:
+    """Test pd.Index[float] / pandas Indexes"""
+    b = pd.Index([True, False, True])
+    i = pd.Index([2, 3, 5])
+    f = pd.Index([1.0, 2.0, 3.0])
+    c = pd.Index([1.1j, 2.2j, 4.1j])
+
+    check(assert_type(left / b, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / i, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / f, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+    check(assert_type(b / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(i / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(f / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(c / left, "pd.Index[complex]"), pd.Index, np.complexfloating)

tests/indexes/arithmetic/int/test_truediv.py

@@ -0,0 +1,103 @@
+import numpy as np
+from numpy import typing as npt  # noqa: F401
+import pandas as pd
+import pytest
+from typing_extensions import assert_type
+
+from tests import (
+    TYPE_CHECKING_INVALID_USAGE,
+    check,
+)
+
+
+@pytest.fixture
+def left() -> "pd.Index[int]":
+    """Left operand"""
+    lo = pd.Index([1, 2, 3])
+    return check(assert_type(lo, "pd.Index[int]"), pd.Index, np.integer)
+
+
+def test_truediv_py_scalar(left: "pd.Index[int]") -> None:
+    """Test pd.Index[int] / Python native scalars"""
+    b, i, f, c = True, 1, 1.0, 1j
+
+    check(assert_type(left / b, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / i, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / f, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+    check(assert_type(b / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(i / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(f / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(c / left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+
+def test_truediv_py_sequence(left: "pd.Index[int]") -> None:
+    """Test pd.Index[int] / Python native sequences"""
+    b, i, f, c = [True, False, True], [2, 3, 5], [1.0, 2.0, 3.0], [1j, 1j, 4j]
+
+    check(assert_type(left / b, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / i, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / f, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+    check(assert_type(b / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(i / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(f / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(c / left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+
+def test_truediv_numpy_array(left: "pd.Index[int]") -> None:
+    """Test pd.Index[int] / numpy arrays"""
+    b = np.array([True, False, True], np.bool_)
+    i = np.array([2, 3, 5], np.int64)
+    f = np.array([1.0, 2.0, 3.0], np.float64)
+    c = np.array([1.1j, 2.2j, 4.1j], np.complex128)
+
+    check(assert_type(left / b, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / i, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / f, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+    # `numpy` typing gives the corresponding `ndarray`s in the static type
+    # checking, where our `__rtruediv__` cannot override. At runtime, they return
+    # `Series` with the correct element type.
+    check(assert_type(b / left, "npt.NDArray[np.float64]"), pd.Index, np.floating)
+    check(assert_type(i / left, "npt.NDArray[np.float64]"), pd.Index, np.floating)
+    check(assert_type(f / left, "npt.NDArray[np.float64]"), pd.Index, np.floating)
+    check(
+        assert_type(c / left, "npt.NDArray[np.complex128]"),
+        pd.Index,
+        np.complexfloating,
+    )
+
+
+def test_truediv_pd_scalar(left: "pd.Index[int]") -> None:
+    """Test pd.Index[int] / pandas scalars"""
+    s, d = pd.Timestamp(2025, 9, 24), pd.Timedelta(seconds=1)
+
+    if TYPE_CHECKING_INVALID_USAGE:
+        _00 = left / s  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+        _01 = left / d  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+
+    if TYPE_CHECKING_INVALID_USAGE:
+        _10 = s / left  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+    check(assert_type(d / left, "pd.Index[pd.Timedelta]"), pd.Index, pd.Timedelta)
+
+
+def test_truediv_pd_index(left: "pd.Index[int]") -> None:
+    """Test pd.Index[int] / pandas Indexes"""
+    b = pd.Index([True, False, True])
+    i = pd.Index([2, 3, 5])
+    f = pd.Index([1.0, 2.0, 3.0])
+    c = pd.Index([1.1j, 2.2j, 4.1j])
+
+    check(assert_type(left / b, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / i, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / f, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(left / c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+    check(assert_type(b / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(i / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(f / left, "pd.Index[float]"), pd.Index, np.floating)
+    check(assert_type(c / left, "pd.Index[complex]"), pd.Index, np.complexfloating)