diff --git a/test/test_api/test_api.py b/test/test_api/test_api.py index 107bd52b..c86be78c 100644 --- a/test/test_api/test_api.py +++ b/test/test_api/test_api.py @@ -134,15 +134,7 @@ def test_infer_on_non_name(Script): def test_infer_on_generator(Script, environment): script = Script('def x(): yield 1\ny=x()\ny') def_, = script.infer() - if environment.version_info >= (3, 9): - # The Generator in Python 3.9 is properly inferred, however once it is - # converted from stub to Python, the definition is - # Generator = _SpecialGenericAlias(collections.abc.Generator, 3) - # This is pretty normal for most typing types, like Sequence, List, - # etc. - assert def_.name == '_SpecialGenericAlias' - else: - assert def_.name == 'Generator' + assert def_.name == 'Generator' def_, = script.infer(only_stubs=True) assert def_.name == 'Generator'