1
0
forked from VimPlug/jedi
Files
jedi-fork/test/test_api/test_usages.py
2020-01-30 19:24:16 +01:00

18 lines
696 B
Python

def test_import_references(Script):
s = Script("from .. import foo", path="foo.py")
assert [usage.line for usage in s.get_references(line=1, column=18)] == [1]
def test_exclude_builtin_modules(Script):
def get(include):
from jedi.api.project import Project
script = Script(source, project=Project('', sys_path=[], smart_sys_path=False))
references = script.get_references(column=8, include_builtins=include)
return [(d.line, d.column) for d in references]
source = '''import sys\nprint(sys.path)'''
places = get(include=True)
assert len(places) > 2 # Includes stubs
places = get(include=False)
assert places == [(1, 7), (2, 6)]