fix(comment): #1397 (comment)

Author: cmp0xff

tests/indexes/arithmetic/test_mul.py

@@ -0,0 +1,112 @@
+from typing import Any
+
+import numpy as np
+import pandas as pd
+import pytest
+from typing_extensions import assert_type
+
+from tests import (
+    TYPE_CHECKING_INVALID_USAGE,
+    check,
+)
+
+
+@pytest.fixture
+def left_i() -> "pd.Index[Any]":
+    """left operand"""
+    lo = pd.MultiIndex.from_tuples([(1,), (2,), (3,)]).levels[0]
+    return check(assert_type(lo, pd.Index), pd.Index)
+
+
+def test_mul_py_scalar(left_i: "pd.Index[Any]") -> None:
+    """Test pd.Index[Any] (int) * Python native scalars"""
+    b, i, f, c = True, 1, 1.0, 1j
+
+    check(assert_type(left_i * b, pd.Index), pd.Index)
+    check(assert_type(left_i * i, pd.Index), pd.Index)
+    check(assert_type(left_i * f, pd.Index), pd.Index)
+    check(assert_type(left_i * c, pd.Index), pd.Index)
+
+    check(assert_type(b * left_i, pd.Index), pd.Index)
+    check(assert_type(i * left_i, pd.Index), pd.Index)
+    check(assert_type(f * left_i, pd.Index), pd.Index)
+    check(assert_type(c * left_i, pd.Index), pd.Index)
+
+
+def test_mul_py_sequence(left_i: "pd.Index[Any]") -> None:
+    """Test pd.Index[Any] (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_i * b, pd.Index), pd.Index)
+    check(assert_type(left_i * i, pd.Index), pd.Index)
+    check(assert_type(left_i * f, pd.Index), pd.Index)
+    check(assert_type(left_i * c, pd.Index), pd.Index)
+
+    check(assert_type(b * left_i, pd.Index), pd.Index)
+    check(assert_type(i * left_i, pd.Index), pd.Index)
+    check(assert_type(f * left_i, pd.Index), pd.Index)
+    check(assert_type(c * left_i, pd.Index), pd.Index)
+
+
+def test_mul_numpy_array(left_i: "pd.Index[Any]") -> None:
+    """Test pd.Index[Any] (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_i * b, pd.Index), pd.Index)
+    check(assert_type(left_i * i, pd.Index), pd.Index)
+    check(assert_type(left_i * f, pd.Index), pd.Index)
+    check(assert_type(left_i * c, pd.Index), pd.Index)
+
+    # `numpy` typing gives the corresponding `ndarray`s in the static type
+    # checking, where our `__rmul__` cannot override. At runtime, they return
+    # `Index`.
+    # microsoft/pyright#10924
+    check(
+        assert_type(b * left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
+    )
+    check(
+        assert_type(i * left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
+    )
+    check(
+        assert_type(f * left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
+    )
+    check(
+        assert_type(c * left_i, Any),  # pyright: ignore[reportAssertTypeFailure]
+        pd.Index,
+    )
+
+
+def test_mul_pd_index(left_i: "pd.Index[Any]") -> None:
+    """Test pd.Index[Any] (int) * pandas Indexes"""
+    a = pd.MultiIndex.from_tuples([(1,), (2,), (3,)]).levels[0]
+    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_i * a, pd.Index), pd.Index)
+    check(assert_type(left_i * b, pd.Index), pd.Index)
+    check(assert_type(left_i * i, pd.Index), pd.Index)
+    check(assert_type(left_i * f, pd.Index), pd.Index)
+    check(assert_type(left_i * c, pd.Index), pd.Index)
+
+    check(assert_type(a * left_i, pd.Index), pd.Index)
+    check(assert_type(b * left_i, pd.Index), pd.Index)
+    check(assert_type(i * left_i, pd.Index), pd.Index)
+    check(assert_type(f * left_i, pd.Index), pd.Index)
+    check(assert_type(c * left_i, pd.Index), pd.Index)
+
+
+def test_mul_str_py_str(left_i: "pd.Index[Any]") -> None:
+    """Test pd.Index[Any] (int) * Python str"""
+    s = "abc"
+
+    if TYPE_CHECKING_INVALID_USAGE:
+        _0 = left_i * s  # type: ignore[operator] # pyright:ignore[reportOperatorIssue]
+        _1 = s * left_i  # type: ignore[operator] # pyright:ignore[reportOperatorIssue]

tests/series/arithmetic/test_mul.py

@@ -2,17 +2,23 @@
 
 import numpy as np
 import pandas as pd
+import pytest
 from typing_extensions import assert_type
 
 from tests import (
     TYPE_CHECKING_INVALID_USAGE,
     check,
 )
 
-left_i = pd.DataFrame({"a": [1, 2, 3]})["a"]  # left operand
 
+@pytest.fixture
+def left_i() -> "pd.Series[Any]":
+    """left operand"""
+    lo = pd.DataFrame({"a": [1, 2, 3]})["a"]
+    return check(assert_type(lo, pd.Series), pd.Series)
 
-def test_mul_py_scalar() -> None:
+
+def test_mul_py_scalar(left_i: "pd.Series[Any]") -> None:
     """Test pd.Series[Any] (int) * Python native scalars"""
     b, i, f, c = True, 1, 1.0, 1j
 
@@ -37,7 +43,7 @@ def test_mul_py_scalar() -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_py_sequence() -> None:
+def test_mul_py_sequence(left_i: "pd.Series[Any]") -> None:
     """Test pd.Series[Any] (int) * Python native sequences"""
     b, i, f, c = [True, False, True], [2, 3, 5], [1.0, 2.0, 3.0], [1j, 1j, 4j]
 
@@ -62,7 +68,7 @@ def test_mul_py_sequence() -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_numpy_array() -> None:
+def test_mul_numpy_array(left_i: "pd.Series[Any]") -> None:
     """Test pd.Series[Any] (int) * numpy arrays"""
     b = np.array([True, False, True], np.bool_)
     i = np.array([2, 3, 5], np.int64)
@@ -106,7 +112,7 @@ def test_mul_numpy_array() -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_pd_index() -> None:
+def test_mul_pd_index(left_i: "pd.Series[Any]") -> None:
     """Test pd.Series[Any] (int) * pandas Indexes"""
     a = pd.MultiIndex.from_tuples([(1,), (2,), (3,)]).levels[0]
     b = pd.Index([True, False, True])
@@ -139,7 +145,7 @@ def test_mul_pd_index() -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_pd_series() -> None:
+def test_mul_pd_series(left_i: "pd.Series[Any]") -> None:
     """Test pd.Series[Any] (int) * pandas Series"""
     a = pd.DataFrame({"a": [1, 2, 3]})["a"]
     b = pd.Series([True, False, True])
@@ -172,7 +178,7 @@ def test_mul_pd_series() -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_str_py_str() -> None:
+def test_mul_str_py_str(left_i: "pd.Series[Any]") -> None:
     """Test pd.Series[Any] (int) * Python str"""
     s = "abc"