fix(comment): #1397 (comment)

Author: cmp0xff

tests/indexes/arithmetic/test_mul.py

@@ -12,13 +12,13 @@
 
 
 @pytest.fixture
-def left_i() -> "pd.Index[Any]":
+def left_i() -> pd.Index:
     """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:
+def test_mul_py_scalar(left_i: pd.Index) -> None:
     """Test pd.Index[Any] (int) * Python native scalars"""
     b, i, f, c = True, 1, 1.0, 1j
 
@@ -33,7 +33,7 @@ def test_mul_py_scalar(left_i: "pd.Index[Any]") -> None:
     check(assert_type(c * left_i, pd.Index), pd.Index)
 
 
-def test_mul_py_sequence(left_i: "pd.Index[Any]") -> None:
+def test_mul_py_sequence(left_i: pd.Index) -> 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]
 
@@ -48,7 +48,7 @@ def test_mul_py_sequence(left_i: "pd.Index[Any]") -> None:
     check(assert_type(c * left_i, pd.Index), pd.Index)
 
 
-def test_mul_numpy_array(left_i: "pd.Index[Any]") -> None:
+def test_mul_numpy_array(left_i: pd.Index) -> None:
     """Test pd.Index[Any] (int) * numpy arrays"""
     b = np.array([True, False, True], np.bool_)
     i = np.array([2, 3, 5], np.int64)
@@ -82,7 +82,7 @@ def test_mul_numpy_array(left_i: "pd.Index[Any]") -> None:
     )
 
 
-def test_mul_pd_index(left_i: "pd.Index[Any]") -> None:
+def test_mul_pd_index(left_i: pd.Index) -> None:
     """Test pd.Index[Any] (int) * pandas Indexes"""
     a = pd.MultiIndex.from_tuples([(1,), (2,), (3,)]).levels[0]
     b = pd.Index([True, False, True])
@@ -103,7 +103,7 @@ def test_mul_pd_index(left_i: "pd.Index[Any]") -> None:
     check(assert_type(c * left_i, pd.Index), pd.Index)
 
 
-def test_mul_str_py_str(left_i: "pd.Index[Any]") -> None:
+def test_mul_str_py_str(left_i: pd.Index) -> None:
     """Test pd.Index[Any] (int) * Python str"""
     s = "abc"
 

tests/series/arithmetic/test_mul.py

@@ -12,13 +12,13 @@
 
 
 @pytest.fixture
-def left_i() -> "pd.Series[Any]":
+def left_i() -> pd.Series:
     """left operand"""
     lo = pd.DataFrame({"a": [1, 2, 3]})["a"]
     return check(assert_type(lo, pd.Series), pd.Series)
 
 
-def test_mul_py_scalar(left_i: "pd.Series[Any]") -> None:
+def test_mul_py_scalar(left_i: pd.Series) -> None:
     """Test pd.Series[Any] (int) * Python native scalars"""
     b, i, f, c = True, 1, 1.0, 1j
 
@@ -43,7 +43,7 @@ def test_mul_py_scalar(left_i: "pd.Series[Any]") -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_py_sequence(left_i: "pd.Series[Any]") -> None:
+def test_mul_py_sequence(left_i: pd.Series) -> 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]
 
@@ -68,7 +68,7 @@ def test_mul_py_sequence(left_i: "pd.Series[Any]") -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_numpy_array(left_i: "pd.Series[Any]") -> None:
+def test_mul_numpy_array(left_i: pd.Series) -> None:
     """Test pd.Series[Any] (int) * numpy arrays"""
     b = np.array([True, False, True], np.bool_)
     i = np.array([2, 3, 5], np.int64)
@@ -112,7 +112,7 @@ def test_mul_numpy_array(left_i: "pd.Series[Any]") -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_pd_index(left_i: "pd.Series[Any]") -> None:
+def test_mul_pd_index(left_i: pd.Series) -> None:
     """Test pd.Series[Any] (int) * pandas Indexes"""
     a = pd.MultiIndex.from_tuples([(1,), (2,), (3,)]).levels[0]
     b = pd.Index([True, False, True])
@@ -145,7 +145,7 @@ def test_mul_pd_index(left_i: "pd.Series[Any]") -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_pd_series(left_i: "pd.Series[Any]") -> None:
+def test_mul_pd_series(left_i: pd.Series) -> None:
     """Test pd.Series[Any] (int) * pandas Series"""
     a = pd.DataFrame({"a": [1, 2, 3]})["a"]
     b = pd.Series([True, False, True])
@@ -178,7 +178,7 @@ def test_mul_pd_series(left_i: "pd.Series[Any]") -> None:
     check(assert_type(left_i.rmul(c), pd.Series), pd.Series)
 
 
-def test_mul_str_py_str(left_i: "pd.Series[Any]") -> None:
+def test_mul_str_py_str(left_i: pd.Series) -> None:
     """Test pd.Series[Any] (int) * Python str"""
     s = "abc"
 

tests/series/test_series.py

@@ -3485,9 +3485,7 @@ def test_diff() -> None:
         )
     # bool -> Any
     check(
-        assert_type(
-            pd.Series([True, True, False, False, True]).diff(), "pd.Series[Any]"
-        ),
+        assert_type(pd.Series([True, True, False, False, True]).diff(), pd.Series),
         pd.Series,
         bool,
         index_to_check_for_type=-1,
@@ -3508,7 +3506,7 @@ def test_diff() -> None:
     )
     # Any -> float
     s_o = s.astype(object)
-    assert_type(s_o, "pd.Series[Any]")
+    assert_type(s_o, pd.Series)
     check(assert_type(s_o.diff(), "pd.Series[float]"), pd.Series, float)
     # complex -> complex
     check(