From 1820aa9476dbc0c05b9d47582ba5017bb444b867 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Fri, 1 May 2026 14:30:00 +0200 Subject: [PATCH] Change some tests slightly --- test/test_api/test_api.py | 2 +- test/test_inference/test_gradual/test_typeshed.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_api/test_api.py b/test/test_api/test_api.py index 7bd1fc88..81c71199 100644 --- a/test/test_api/test_api.py +++ b/test/test_api/test_api.py @@ -134,7 +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() - assert def_.name == 'Generator' + assert def_.name == 'GeneratorType' def_, = script.infer(only_stubs=True) assert def_.name == 'Generator' diff --git a/test/test_inference/test_gradual/test_typeshed.py b/test/test_inference/test_gradual/test_typeshed.py index 0f9ffead..0b1054a2 100644 --- a/test/test_inference/test_gradual/test_typeshed.py +++ b/test/test_inference/test_gradual/test_typeshed.py @@ -75,7 +75,7 @@ def test_sys_exc_info(Script): # It's an optional. assert def_.name == 'BaseException' assert def_.module_path == typeshed.TYPESHED_PATH.joinpath( - 'stdlib', '3', 'builtins.pyi' + 'stdlib', 'builtins.pyi' ) assert def_.type == 'instance' assert none.name == 'NoneType'