Use a class stub class

This commit is contained in:
Dave Halter
2018-08-01 10:47:46 +02:00
parent c196075cb8
commit c8caa8f4ac
2 changed files with 16 additions and 5 deletions

View File

@@ -37,14 +37,20 @@ def test_get_stub_files():
def test_function(Script):
s = Script('import threading; threading.current_thread')
def_, = s.goto_definitions()
code = 'import threading; threading.current_thread'
def_, = Script(code).goto_definitions()
context = def_._name._context
assert isinstance(context, typeshed.FunctionStubContext), context
def_, = Script(code + '()').goto_definitions()
context = def_._name._context
assert isinstance(context, typeshed.ClassStubContext), context
def test_class(Script):
s = Script('import threading; threading.Lock')
def_, = Script('import threading; threading.Lock').goto_definitions()
context = def_._name._context
assert isinstance(context, typeshed.ClassStubContext), context
def test_instance(Script):