wrapping generics in quotes for compatibility

muddi900 · muddi900 · commit a4013d2e7441 · 2023-01-03T21:52:37.000+05:00
diff --git a/tests/test_frame.py b/tests/test_frame.py
@@ -1806,12 +1806,12 @@ def test_frame_stack() -> None:
     )
 
     check(
-        assert_type(df_multi_level_cols2.stack(0), Union[pd.DataFrame, pd.Series[Any]]),
+        assert_type(df_multi_level_cols2.stack(0), Union[pd.DataFrame, "pd.Series[Any]"]),
         pd.DataFrame,
     )
     check(
         assert_type(
-            df_multi_level_cols2.stack([0, 1]), Union[pd.DataFrame, pd.Series[Any]]
+            df_multi_level_cols2.stack([0, 1]), Union[pd.DataFrame, "pd.Series[Any]"]
         ),
         pd.Series,
     )

Original file line number	Diff line number	Diff line change
`@@ -1806,12 +1806,12 @@ def test_frame_stack() -> None:`
`1806`	`1806`	`)`
`1807`	`1807`
`1808`	`1808`	`check(`
`1809`		`- assert_type(df_multi_level_cols2.stack(0), Union[pd.DataFrame, pd.Series[Any]]),`
	`1809`	`+ assert_type(df_multi_level_cols2.stack(0), Union[pd.DataFrame, "pd.Series[Any]"]),`
`1810`	`1810`	`pd.DataFrame,`
`1811`	`1811`	`)`
`1812`	`1812`	`check(`
`1813`	`1813`	`assert_type(`
`1814`		`- df_multi_level_cols2.stack([0, 1]), Union[pd.DataFrame, pd.Series[Any]]`
	`1814`	`+ df_multi_level_cols2.stack([0, 1]), Union[pd.DataFrame, "pd.Series[Any]"]`
`1815`	`1815`	`),`
`1816`	`1816`	`pd.Series,`
`1817`	`1817`	`)`