mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 22:14:27 +08:00
24 lines
805 B
Python
24 lines
805 B
Python
import pytest
|
|
|
|
from jedi import settings
|
|
from jedi.evaluate.filters import ContextName
|
|
from jedi.evaluate.compiled import CompiledContextName
|
|
from jedi.evaluate.gradual.typeshed import StubOnlyModuleContext
|
|
|
|
|
|
@pytest.fixture()
|
|
def auto_import_json(monkeypatch):
|
|
monkeypatch.setattr(settings, 'auto_import_modules', ['json'])
|
|
|
|
|
|
def test_base_auto_import_modules(auto_import_json, Script):
|
|
loads, = Script('import json; json.loads').goto_definitions()
|
|
assert isinstance(loads._name, ContextName)
|
|
context, = loads._name.infer()
|
|
assert isinstance(context.parent_context, StubOnlyModuleContext)
|
|
|
|
|
|
def test_auto_import_modules_imports(auto_import_json, Script):
|
|
main, = Script('from json import tool; tool.main').goto_definitions()
|
|
assert isinstance(main._name, CompiledContextName)
|