mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-07 06:24:27 +08:00
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
import os
|
|
|
|
from jedi.plugins import typeshed
|
|
from parso.utils import PythonVersionInfo
|
|
|
|
|
|
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)
|
|
|
|
base = os.path.join(typeshed._TYPESHED_PATH, 'stdlib', '3')
|
|
map_ = typeshed._create_stub_map(base)
|
|
assert map_['functools'] == os.path.join(base, 'functools.pyi')
|