VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

actually use auto_import_modules correctly

Browse Source
This commit is contained in:
Dave Halter
2021-01-01 02:59:42 +01:00
parent c64e33173a
commit 92d96ac336
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jedi/inference/gradual/typeshed.py
View File

@@ -120,7 +120,7 @@ def import_module_decorator(func):
) )
inference_state.module_cache.add(import_names, python_value_set) inference_state.module_cache.add(import_names, python_value_set)
if not prefer_stubs: if not prefer_stubs or import_names[0] in settings.auto_import_modules:
return python_value_set return python_value_set
stub = try_to_load_stub_cached(inference_state, import_names, python_value_set, stub = try_to_load_stub_cached(inference_state, import_names, python_value_set,

7
test/test_settings.py
View File

@@ -1,9 +1,8 @@
import pytest import pytest
from jedi import settings from jedi import settings
from jedi.inference.names import ValueName
from jedi.inference.compiled import CompiledValueName from jedi.inference.compiled import CompiledValueName
from jedi.inference.gradual.typeshed import StubModuleValue from jedi.inference.compiled.value import CompiledModule
@pytest.fixture() @pytest.fixture()
@@ -13,9 +12,9 @@ def auto_import_json(monkeypatch):
def test_base_auto_import_modules(auto_import_json, Script): def test_base_auto_import_modules(auto_import_json, Script):
loads, = Script('import json; json.loads').infer() loads, = Script('import json; json.loads').infer()
assert isinstance(loads._name, ValueName) assert isinstance(loads._name, CompiledValueName)
value, = loads._name.infer() value, = loads._name.infer()
assert isinstance(value.parent_context._value, StubModuleValue) assert isinstance(value.parent_context._value, CompiledModule)
def test_auto_import_modules_imports(auto_import_json, Script): def test_auto_import_modules_imports(auto_import_json, Script):