mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 14:04:26 +08:00
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
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')
|
|
|
|
|
|
def test_get_typeshed_directories():
|
|
def get_dirs(version_info):
|
|
return {
|
|
d.replace(typeshed._TYPESHED_PATH, '').lstrip(os.path.sep)
|
|
for d in typeshed._get_typeshed_directories(version_info)
|
|
}
|
|
|
|
dirs = get_dirs(PythonVersionInfo(2, 7))
|
|
assert dirs == {'stdlib/2and3', 'stdlib/2', 'third_party/2and3', 'third_party/2'}
|
|
|
|
dirs = get_dirs(PythonVersionInfo(3, 4))
|
|
assert dirs == {'stdlib/2and3', 'stdlib/3', 'third_party/2and3', 'third_party/3'}
|
|
|
|
dirs = get_dirs(PythonVersionInfo(3, 5))
|
|
assert dirs == {'stdlib/2and3', 'stdlib/3', 'stdlib/3.5',
|
|
'third_party/2and3', 'third_party/3', 'third_party/3.5'}
|
|
|
|
dirs = get_dirs(PythonVersionInfo(3, 6))
|
|
assert dirs == {'stdlib/2and3', 'stdlib/3', 'stdlib/3.5', 'stdlib/3.6',
|
|
'third_party/2and3', 'third_party/3', 'third_party/3.5', 'third_party/3.6'}
|
|
|
|
|
|
def test_get_stub_files():
|
|
def get_map(version_info):
|
|
return typeshed._create_stub_map(version_info)
|
|
|
|
map_ = typeshed._create_stub_map(TYPESHED_PYTHON3)
|
|
assert map_['functools'] == os.path.join(TYPESHED_PYTHON3, 'functools.pyi')
|
|
|
|
|
|
def test_function(Script):
|
|
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, TreeInstance)
|
|
assert isinstance(context.class_context, typeshed.ClassStubContext), context
|
|
|
|
|
|
def test_class(Script):
|
|
def_, = Script('import threading; threading.Thread').goto_definitions()
|
|
context = def_._name._context
|
|
assert isinstance(context, typeshed.ClassStubContext), context
|
|
|
|
|
|
def test_instance(Script):
|
|
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_, = Script('import threading; threading.Thread.getName').goto_definitions()
|
|
context = def_._name._context
|
|
assert isinstance(context, typeshed.FunctionStubContext), context
|
|
|
|
|
|
def test_method(Script):
|
|
code = 'import threading; threading.Thread().getName'
|
|
def_, = Script(code).goto_definitions()
|
|
context = def_._name._context
|
|
assert isinstance(context, typeshed.ClassStubContext), context
|