From 8e3caaca7f436bd008c6e1ae81507e0218d85962 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Sat, 15 Jun 2019 02:20:15 +0200 Subject: [PATCH] Improve the stub test a bit --- test/test_evaluate/test_gradual/test_stubs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_evaluate/test_gradual/test_stubs.py b/test/test_evaluate/test_gradual/test_stubs.py index 4e58c881..63442676 100644 --- a/test/test_evaluate/test_gradual/test_stubs.py +++ b/test/test_evaluate/test_gradual/test_stubs.py @@ -75,5 +75,5 @@ def test_infer_and_goto(Script, code, full_name, has_stub, has_python, way, if only_stubs: assert d.is_stub() assert d.full_name == full_name - #if d.is_stub(): - #assert d.module_path.endswith('.pyi') + + assert d.is_stub() == d.module_path.endswith('.pyi')