complex

Author: cmp0xff

pandas-stubs/core/indexes/base.pyi

@@ -752,6 +752,8 @@ class Index(IndexOpsMixin[S1]):
     def __mul__(self, other: Index[Never]) -> Index: ...
     @overload
     def __mul__(self, other: np_ndarray_dt) -> Never: ...
+    @overload
+    def __mul__(self: Index[complex], other: np_ndarray_td) -> Never: ...
     # pandas-dev/pandas#62524
     @overload
     def __mul__(  # type: ignore[overload-overlap]
@@ -823,6 +825,8 @@ class Index(IndexOpsMixin[S1]):
     def __rmul__(self, other: Index[Never]) -> Index: ...
     @overload
     def __rmul__(self, other: np_ndarray_dt) -> Never: ...
+    @overload
+    def __rmul__(self: Index[complex], other: np_ndarray_td) -> Never: ...
     # pandas-dev/pandas#62524
     @overload
     def __rmul__(  # type: ignore[overload-overlap]

pandas-stubs/core/series.pyi

@@ -2548,6 +2548,8 @@ class Series(IndexOpsMixin[S1], NDFrame):
     @overload
     def __mul__(self, other: np_ndarray_dt) -> Never: ...
     @overload
+    def __mul__(self: Series[complex], other: np_ndarray_td) -> Never: ...
+    @overload
     def __mul__(
         self: Series[bool] | Series[int] | Series[float],
         other: (
@@ -2777,6 +2779,8 @@ class Series(IndexOpsMixin[S1], NDFrame):
     @overload
     def __rmul__(self: Series[Timestamp], other: np_ndarray) -> Never: ...
     @overload
+    def __rmul__(self: Series[complex], other: np_ndarray_td) -> Never: ...
+    @overload
     def __rmul__(self: Series[Timedelta], other: np_ndarray_complex) -> Never: ...
     @overload
     def __rmul__(

tests/indexes/arithmetic/complex/test_mul.py

@@ -0,0 +1,132 @@
+from datetime import (
+    datetime,
+    timedelta,
+)
+from typing import Any
+
+import numpy as np
+from numpy import typing as npt  # noqa: F401
+import pandas as pd
+import pytest
+from typing_extensions import (
+    Never,
+    assert_type,
+)
+
+from tests import (
+    TYPE_CHECKING_INVALID_USAGE,
+    check,
+)
+
+
+@pytest.fixture
+def left() -> "pd.Index[complex]":
+    """left operand"""
+    lo = pd.Index([1j, 2j, 3j])
+    return check(assert_type(lo, "pd.Index[complex]"), pd.Index, np.complexfloating)
+
+
+def test_mul_py_scalar(left: "pd.Index[complex]") -> None:
+    """Test pd.Index[complex] * Python native scalars"""
+    b, i, f, c = True, 1, 1.0, 1j
+    s, d = datetime(2025, 9, 30), timedelta(seconds=1)
+
+    check(assert_type(left * b, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * i, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * f, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    if TYPE_CHECKING_INVALID_USAGE:
+        _05 = left * s  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+        _06 = left * d  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+
+    check(assert_type(b * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(i * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(f * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(c * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    if TYPE_CHECKING_INVALID_USAGE:
+        _15 = s * left  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+        _16 = d * left  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+
+
+def test_mul_py_sequence(left: "pd.Index[complex]") -> None:
+    """Test pd.Index[complex] * Python native sequences"""
+    b, i, f, c = [True, False, True], [2, 3, 5], [1.0, 2.0, 3.0], [1j, 1j, 4j]
+    s = [datetime(2025, 9, d) for d in (27, 28, 29)]
+    d = [timedelta(seconds=s + 1) for s in range(3)]
+
+    check(assert_type(left * b, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * i, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * f, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    if TYPE_CHECKING_INVALID_USAGE:
+        _05 = left * s  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+        _06 = left * d  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+
+    check(assert_type(b * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(i * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(f * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(c * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    if TYPE_CHECKING_INVALID_USAGE:
+        _15 = s * left  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+        _16 = d * left  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+
+
+def test_mul_numpy_array(left: "pd.Index[complex]") -> None:
+    """Test pd.Index[complex] * 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)
+    s = np.array([np.datetime64(f"2025-09-{d}") for d in (27, 28, 29)], np.datetime64)
+    d = np.array([np.timedelta64(s + 1, "s") for s in range(3)], np.timedelta64)
+
+    check(assert_type(left * b, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * i, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * f, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    if TYPE_CHECKING_INVALID_USAGE:
+        assert_type(left * s, Never)
+        assert_type(left * d, Never)
+
+    # `numpy` typing gives the corresponding `ndarray`s in the static type
+    # checking, where our `__rmul__` cannot override. At runtime, they return
+    # `Index` with the correct element type.
+    check(assert_type(b * left, "npt.NDArray[np.bool_]"), pd.Index, np.complexfloating)
+    check(assert_type(i * left, "npt.NDArray[np.int64]"), pd.Index, np.complexfloating)
+    check(
+        assert_type(f * left, "npt.NDArray[np.float64]"), pd.Index, np.complexfloating
+    )
+    check(
+        assert_type(c * left, "npt.NDArray[np.complex128]"),
+        pd.Index,
+        np.complexfloating,
+    )
+    if TYPE_CHECKING_INVALID_USAGE:
+        assert_type(s * left, Any)
+        assert_type(d * left, "npt.NDArray[np.timedelta64]")
+
+
+def test_mul_pd_index(left: "pd.Index[complex]") -> None:
+    """Test pd.Index[complex] * 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])
+    s = pd.Index([datetime(2025, 9, d) for d in (27, 28, 29)])
+    d = pd.Index([timedelta(seconds=s + 1) for s in range(3)])
+
+    check(assert_type(left * b, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * i, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * f, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(left * c, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    if TYPE_CHECKING_INVALID_USAGE:
+        _05 = left * s  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+        _06 = left * d  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+
+    check(assert_type(b * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(i * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(f * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    check(assert_type(c * left, "pd.Index[complex]"), pd.Index, np.complexfloating)
+    if TYPE_CHECKING_INVALID_USAGE:
+        _15 = s * left  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
+        _16 = d * left  # type: ignore[operator] # pyright: ignore[reportOperatorIssue]

tests/indexes/arithmetic/datetime/__init__.py renamed to tests/indexes/arithmetic/datetimeindex/__init__.py

@@ -90,7 +90,7 @@ def test_mul_numpy_array(left: "pd.Index[float]") -> None:
 
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__rmul__` cannot override. At runtime, they return
-    # `Series` with the correct element type.
+    # `Index` with the correct element type.
     check(assert_type(b * left, "npt.NDArray[np.bool_]"), pd.Index, np.floating)
     check(assert_type(i * left, "npt.NDArray[np.int64]"), pd.Index, np.floating)
     check(assert_type(f * left, "npt.NDArray[np.float64]"), pd.Index, np.floating)

tests/indexes/arithmetic/float/test_mul.py

@@ -92,7 +92,7 @@ def test_mul_numpy_array(left: "pd.Index[int]") -> None:
 
     # `numpy` typing gives the corresponding `ndarray`s in the static type
     # checking, where our `__rmul__` cannot override. At runtime, they return
-    # `Series` with the correct element type.
+    # `Index` with the correct element type.
     check(assert_type(b * left, "npt.NDArray[np.bool_]"), pd.Index, np.integer)
     check(assert_type(i * left, "npt.NDArray[np.int64]"), pd.Index, np.integer)
     check(assert_type(f * left, "npt.NDArray[np.float64]"), pd.Index, np.floating)