Fix some name inferance with stubs

This commit is contained in:
Dave Halter
2018-09-05 10:29:37 +02:00
parent 39162de2a8
commit fa16c9e59d
5 changed files with 21 additions and 7 deletions

View File

@@ -1,7 +1,10 @@
import pytest
from jedi import settings
from jedi.evaluate.filters import ContextName
from jedi.evaluate.compiled import CompiledContextName
from jedi.plugins.typeshed import StubContextWithCompiled
from jedi.evaluate.context import FunctionContext
@pytest.fixture()
@@ -11,7 +14,10 @@ def auto_import_json(monkeypatch):
def test_base_auto_import_modules(auto_import_json, Script):
loads, = Script('import json; json.loads').goto_definitions()
assert isinstance(loads._name, CompiledContextName)
assert isinstance(loads._name, ContextName)
context, = loads._name.infer()
assert isinstance(context, StubContextWithCompiled)
assert isinstance(context._wrapped_context, FunctionContext)
def test_auto_import_modules_imports(auto_import_json, Script):