Index[complex]

Author: cmp0xff

tests/indexes/arithmetic/complex/test_truediv.py

@@ -0,0 +1,93 @@
+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 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_truediv_py_scalar(left: "pd.Index[complex]") -> None:
+    """Test pd.Index[complex] / Python native scalars"""
+    b, i, f, c = True, 1, 1.0, 1j
+
+    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)
+
+    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)
+
+
+def test_truediv_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]
+
+    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)
+
+    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)
+
+
+def test_truediv_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)
+
+    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)
+
+    # `numpy` typing gives the corresponding `ndarray`s in the static type
+    # checking, where our `__rtruediv__` cannot override. At runtime, they return
+    # `Index` with the correct element type.
+    check(
+        assert_type(b / left, "npt.NDArray[np.float64]"), pd.Index, np.complexfloating
+    )
+    check(
+        assert_type(i / left, "npt.NDArray[np.float64]"), 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,
+    )
+
+
+def test_truediv_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])
+
+    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)
+
+    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)

tests/indexes/arithmetic/test_truediv.py

@@ -114,8 +114,7 @@ def test_truediv_paths(tmp_path: Path) -> None:
 
     check(assert_type(folders / fpath, pd.Index), pd.Index, Path)
 
-    # mypy thinks it's `Path`, in contrast to Index.__rtruediv__(self, other: Path) -> Index: ...
-    check(assert_type(tmp_path / fpaths, pd.Index), pd.Index, Path)  # type: ignore[assert-type]
+    check(assert_type(tmp_path / fpaths, pd.Index), pd.Index, Path)
 
 
 def test_truediv_path(tmp_path: Path) -> None:

tests/series/arithmetic/complex/test_truediv.py

@@ -1,14 +1,20 @@
 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 check
 
-left = pd.Series([1j, 2j, 3j])  # left operand
 
+@pytest.fixture
+def left() -> "pd.Series[complex]":
+    """Left operand"""
+    lo = pd.Series([1j, 2j, 3j])
+    return check(assert_type(lo, "pd.Series[complex]"), pd.Series, np.complexfloating)
 
-def test_truediv_py_scalar() -> None:
+
+def test_truediv_py_scalar(left: "pd.Series[complex]") -> None:
     """Test pd.Series[complex] / Python native scalars"""
     b, i, f, c = True, 1, 1.0, 1j
 
@@ -83,7 +89,7 @@ def test_truediv_py_scalar() -> None:
     )
 
 
-def test_truediv_py_sequence() -> None:
+def test_truediv_py_sequence(left: "pd.Series[complex]") -> None:
     """Test pd.Series[complex] / Python native sequences"""
     b, i, f, c = [True, False, True], [2, 3, 5], [1.0, 2.0, 3.0], [1j, 1j, 4j]
 
@@ -158,7 +164,7 @@ def test_truediv_py_sequence() -> None:
     )
 
 
-def test_truediv_numpy_array() -> None:
+def test_truediv_numpy_array(left: "pd.Series[complex]") -> None:
     """Test pd.Series[complex] / numpy arrays"""
     b = np.array([True, False, True], np.bool_)
     i = np.array([2, 3, 5], np.int64)
@@ -249,7 +255,7 @@ def test_truediv_numpy_array() -> None:
     )
 
 
-def test_truediv_pd_index() -> None:
+def test_truediv_pd_index(left: "pd.Series[complex]") -> None:
     """Test pd.Series[complex] / pandas Indexes"""
     b = pd.Index([True, False, True])
     i = pd.Index([2, 3, 5])
@@ -327,7 +333,7 @@ def test_truediv_pd_index() -> None:
     )
 
 
-def test_truediv_pd_series() -> None:
+def test_truediv_pd_series(left: "pd.Series[complex]") -> None:
     """Test pd.Series[complex] / pandas Series"""
     b = pd.Series([True, False, True])
     i = pd.Series([2, 3, 5])