1
0
forked from VimPlug/jedi

actually use auto_import_modules correctly

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

View File

@@ -120,7 +120,7 @@ def import_module_decorator(func):
)
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
stub = try_to_load_stub_cached(inference_state, import_names, python_value_set,

View File

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