unquote pd.Series in tests

Author: MarcoGorelli

pandas-stubs/core/reshape/concat.pyi

@@ -126,7 +126,7 @@ def concat(
 ) -> DataFrame: ...
 
 # Including either of the next 2 overloads causes mypy to complain about
-# test_pandas.py:test_types_concat() in assert_type(pd.concat([s, s2]), "pd.Series")
+# test_pandas.py:test_types_concat() in assert_type(pd.concat([s, s2]), pd.Series)
 # It thinks that pd.concat([s, s2]) is Any .  May be due to Series being
 # Generic, or the axis argument being unspecified, and then there is partial
 # overlap with the first 2 overloads.

tests/test_frame.py

@@ -682,8 +682,8 @@ def test_frame_iterator() -> None:
     """Test iterator methods for a dataframe GH1217."""
     df = pd.DataFrame(data={"col1": [2, 1], "col2": [3, 4]})
 
-    check(assert_type(next(df.items()), tuple[Hashable, "pd.Series"]), tuple)
-    check(assert_type(next(df.iterrows()), tuple[Hashable, "pd.Series"]), tuple)
+    check(assert_type(next(df.items()), tuple[Hashable, pd.Series]), tuple)
+    check(assert_type(next(df.iterrows()), tuple[Hashable, pd.Series]), tuple)
     check(assert_type(next(df.itertuples()), _PandasNamedTuple), _PandasNamedTuple)
 
 
@@ -3101,30 +3101,28 @@ def test_frame_stack() -> None:
         upper="2.2.99",
     ):
         check(
-            assert_type(
-                df_multi_level_cols2.stack(0), Union[pd.DataFrame, "pd.Series"]
-            ),
+            assert_type(df_multi_level_cols2.stack(0), Union[pd.DataFrame, pd.Series]),
             pd.DataFrame,
         )
         check(
             assert_type(
                 df_multi_level_cols2.stack([0, 1]),
-                Union[pd.DataFrame, "pd.Series"],
+                Union[pd.DataFrame, pd.Series],
             ),
             pd.Series,
         )
         if PD_LTE_22:
             check(
                 assert_type(
                     df_multi_level_cols2.stack(0, future_stack=False),
-                    Union[pd.DataFrame, "pd.Series"],
+                    Union[pd.DataFrame, pd.Series],
                 ),
                 pd.DataFrame,
             )
             check(
                 assert_type(
                     df_multi_level_cols2.stack(0, dropna=True, sort=True),
-                    Union[pd.DataFrame, "pd.Series"],
+                    Union[pd.DataFrame, pd.Series],
                 ),
                 pd.DataFrame,
             )

tests/test_series.py

@@ -2872,13 +2872,13 @@ def test_astype_object(cast_arg: ObjectDtypeArg, target_type: type) -> None:
 
     if TYPE_CHECKING:
         # python object
-        assert_type(s.astype(object), "pd.Series")
-        assert_type(s.astype("object"), "pd.Series")
+        assert_type(s.astype(object), pd.Series)
+        assert_type(s.astype("object"), pd.Series)
         # numpy object
-        assert_type(s.astype(np.object_), "pd.Series")
-        # assert_type(s.astype("object_"), "pd.Series")  # NOTE: not assigned
-        # assert_type(s.astype("object0"), "pd.Series")  # NOTE: not assigned
-        assert_type(s.astype("O"), "pd.Series")
+        assert_type(s.astype(np.object_), pd.Series)
+        # assert_type(s.astype("object_"), pd.Series)  # NOTE: not assigned
+        # assert_type(s.astype("object0"), pd.Series)  # NOTE: not assigned
+        assert_type(s.astype("O"), pd.Series)
 
 
 @pytest.mark.parametrize("cast_arg, target_type", ASTYPE_VOID_ARGS, ids=repr)
@@ -2888,9 +2888,9 @@ def test_astype_void(cast_arg: VoidDtypeArg, target_type: type) -> None:
 
     if TYPE_CHECKING:
         # numpy void
-        assert_type(s.astype(np.void), "pd.Series")
-        assert_type(s.astype("void"), "pd.Series")
-        assert_type(s.astype("V"), "pd.Series")
+        assert_type(s.astype(np.void), pd.Series)
+        assert_type(s.astype("void"), pd.Series)
+        assert_type(s.astype("V"), pd.Series)
 
 
 def test_astype_other() -> None:
@@ -2902,7 +2902,7 @@ def test_astype_other() -> None:
 
     # Test self-consistent with s.dtype (#747)
     # NOTE: https://github.com/python/typing/issues/801#issuecomment-1646171898
-    check(assert_type(s.astype(s.dtype), "pd.Series"), pd.Series, np.integer)
+    check(assert_type(s.astype(s.dtype), pd.Series), pd.Series, np.integer)
 
     # test DecimalDtype
     orseries = pd.Series([Decimal(x) for x in [1, 2, 3]])
@@ -2917,7 +2917,7 @@ def test_astype_other() -> None:
     # Test non-literal string
     # NOTE: currently unsupported! Enable in future.
     # string: str = "int"  # not Literal!
-    # check(assert_type(s.astype(string), "pd.Series"), pd.Series, np.integer)
+    # check(assert_type(s.astype(string), pd.Series), pd.Series, np.integer)
 
 
 def test_all_astype_args_tested() -> None:
@@ -3224,7 +3224,7 @@ def test_get() -> None:
 
 def test_series_new_empty() -> None:
     # GH 826
-    check(assert_type(pd.Series(), "pd.Series"), pd.Series)
+    check(assert_type(pd.Series(), pd.Series), pd.Series)
 
 
 def test_series_mapping() -> None:
@@ -3406,10 +3406,10 @@ def first_arg_not_series(argument_1: int, ser: pd.Series) -> pd.Series:
 
 def test_series_apply() -> None:
     s = pd.Series(["A", "B", "AB"])
-    check(assert_type(s.apply(tuple), "pd.Series"), pd.Series)
-    check(assert_type(s.apply(list), "pd.Series"), pd.Series)
-    check(assert_type(s.apply(set), "pd.Series"), pd.Series)
-    check(assert_type(s.apply(frozenset), "pd.Series"), pd.Series)
+    check(assert_type(s.apply(tuple), pd.Series), pd.Series)
+    check(assert_type(s.apply(list), pd.Series), pd.Series)
+    check(assert_type(s.apply(set), pd.Series), pd.Series)
+    check(assert_type(s.apply(frozenset), pd.Series), pd.Series)
 
 
 def test_diff() -> None:
@@ -3571,7 +3571,7 @@ def callable(x: int) -> str:
 
     unknown_series = pd.Series([1, 0, None])
     check(
-        assert_type(unknown_series.map({1: True, 0: False, None: None}), "pd.Series"),
+        assert_type(unknown_series.map({1: True, 0: False, None: None}), pd.Series),
         pd.Series,
     )
 
@@ -3673,8 +3673,8 @@ class MyDict(TypedDict):
 def test_series_empty_dtype() -> None:
     """Test for the creation of a Series from an empty list GH571 to map to a Series."""
     new_tab: Sequence[Never] = []  # need to be typehinted to please mypy
-    check(assert_type(pd.Series(new_tab), "pd.Series"), pd.Series)
-    check(assert_type(pd.Series([]), "pd.Series"), pd.Series)
+    check(assert_type(pd.Series(new_tab), pd.Series), pd.Series)
+    check(assert_type(pd.Series([]), pd.Series), pd.Series)
     # ensure that an empty string does not get matched to Sequence[Never]
     check(assert_type(pd.Series(""), "pd.Series[str]"), pd.Series)