From 31bf8e48bb02231d39e38c95e7c47cc21157b5aa Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Thu, 2 Aug 2018 00:59:12 +0200 Subject: [PATCH] Fix some stub tests --- test/test_plugin/test_stub.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/test/test_plugin/test_stub.py b/test/test_plugin/test_stub.py index d84a865b..5b4d337d 100644 --- a/test/test_plugin/test_stub.py +++ b/test/test_plugin/test_stub.py @@ -1,6 +1,7 @@ import os from jedi.plugins import typeshed +from jedi.evaluate.context import TreeInstance from parso.utils import PythonVersionInfo TYPESHED_PYTHON3 = os.path.join(typeshed._TYPESHED_PATH, 'stdlib', '3') @@ -44,7 +45,8 @@ def test_function(Script): def_, = Script(code + '()').goto_definitions() context = def_._name._context - assert isinstance(context, typeshed.ClassStubContext), context + assert isinstance(context, TreeInstance) + assert isinstance(context.class_context, typeshed.ClassStubContext), context def test_class(Script): @@ -54,7 +56,10 @@ def test_class(Script): def test_instance(Script): - s = Script('import threading; threading.Thread()') + def_, = Script('import threading; threading.Thread()').goto_definitions() + context = def_._name._context + assert isinstance(context, TreeInstance) + assert isinstance(context.class_context, typeshed.ClassStubContext), context def test_class_function(Script):