Fix some stub tests

This commit is contained in:
Dave Halter
2018-08-02 00:59:12 +02:00
parent 61de28f741
commit 31bf8e48bb

View File

@@ -1,6 +1,7 @@
import os import os
from jedi.plugins import typeshed from jedi.plugins import typeshed
from jedi.evaluate.context import TreeInstance
from parso.utils import PythonVersionInfo from parso.utils import PythonVersionInfo
TYPESHED_PYTHON3 = os.path.join(typeshed._TYPESHED_PATH, 'stdlib', '3') TYPESHED_PYTHON3 = os.path.join(typeshed._TYPESHED_PATH, 'stdlib', '3')
@@ -44,7 +45,8 @@ def test_function(Script):
def_, = Script(code + '()').goto_definitions() def_, = Script(code + '()').goto_definitions()
context = def_._name._context 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): def test_class(Script):
@@ -54,7 +56,10 @@ def test_class(Script):
def test_instance(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): def test_class_function(Script):