GH1055 Fix tests

Author: loicdiridollou

pandas-stubs/api/typing/__init__.pyi

@@ -2,7 +2,6 @@ from pandas.core.groupby import (
     DataFrameGroupBy as DataFrameGroupBy,
     SeriesGroupBy as SeriesGroupBy,
 )
-from pandas.core.indexes.frozen import FrozenList as FrozenList
 from pandas.core.resample import (
     DatetimeIndexResamplerGroupby as DatetimeIndexResamplerGroupby,
     PeriodIndexResamplerGroupby as PeriodIndexResamplerGroupby,

tests/test_api_typing.py

@@ -1,22 +1,19 @@
 """Test module for classes in pandas.api.typing."""
 
-from typing import (
-    TYPE_CHECKING,
-    Literal,
-)
+from typing import Literal  # noqa: F401
+from typing import TYPE_CHECKING
 
 import numpy as np
 import pandas as pd
 from pandas.api.typing import (
     DataFrameGroupBy,
     DatetimeIndexResamplerGroupby,
-    FrozenList,
     NAType,
     SeriesGroupBy,
 )
 from typing_extensions import assert_type
 
-from pandas._typing import Scalar
+from pandas._typing import Scalar  # noqa: F401
 
 from tests import check
 
@@ -27,24 +24,19 @@
 def test_dataframegroupby():
     df = pd.DataFrame({"a": [1, 2, 3]})
     check(
-        assert_type(df.groupby("a"), DataFrameGroupBy[Scalar, Literal[True]]),
+        assert_type(df.groupby("a"), "DataFrameGroupBy[Scalar, Literal[True]]"),
         DataFrameGroupBy,
     )
 
 
 def test_seriesgroupby():
     sr: pd.Series[int] = pd.Series([1, 2, 3], index=pd.Index(["a", "b", "a"]))
     check(
-        assert_type(sr.groupby("a"), SeriesGroupBy[int, Scalar]),
+        assert_type(sr.groupby(level=0), "SeriesGroupBy[int, Scalar]"),
         SeriesGroupBy,
     )
 
 
-def test_frozenlist() -> None:
-    flst = FrozenList([1, 2, 3])
-    check(assert_type(flst, FrozenList), FrozenList)
-
-
 def tests_datetimeindexersamplergroupby() -> None:
     idx = pd.date_range("1999-1-1", periods=365, freq="D")
     df = pd.DataFrame(