forked from VimPlug/jedi
93 lines
2.5 KiB
Python
93 lines
2.5 KiB
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)]
|
|
|
|
|
|
def test_references_scope(Script):
|
|
from jedi.api.project import Project
|
|
project = Project('', sys_path=[], smart_sys_path=False)
|
|
script = Script(
|
|
'''import sys
|
|
from collections import defaultdict
|
|
|
|
print(sys.path)
|
|
|
|
def foo(bar):
|
|
baz = defaultdict(int)
|
|
return baz
|
|
|
|
def bar(foo):
|
|
baz = defaultdict(int)
|
|
return baz
|
|
|
|
foo()
|
|
''', project=project)
|
|
|
|
def r(*args):
|
|
return script.get_references(scope='file', *args)
|
|
|
|
print(script._code_lines)
|
|
sys_places = r(1, 7)
|
|
assert len(sys_places) == 2
|
|
assert sys_places == r(4, 6)
|
|
|
|
assert len(r(2, 5)) == 1
|
|
|
|
dd_places = r(2, 24)
|
|
assert len(dd_places) == 3
|
|
assert dd_places == r(7, 10)
|
|
assert dd_places == r(11, 10)
|
|
|
|
foo_places = r(6, 4)
|
|
assert len(foo_places) == 2
|
|
assert foo_places == r(14, 0)
|
|
|
|
baz_places = r(7, 4)
|
|
assert len(baz_places) == 2
|
|
assert baz_places == r(8, 11)
|
|
|
|
int_places = r(7, 22)
|
|
assert len(int_places) == 2
|
|
assert int_places == r(11, 22)
|
|
|
|
baz_places = r(11, 4)
|
|
assert len(baz_places) == 2
|
|
assert baz_places == r(12, 11)
|
|
|
|
script = Script('from datetime', project=project)
|
|
places = r(1, 5)
|
|
assert len(places) == 1
|
|
|
|
|
|
def test_local_references_method_other_file(Script):
|
|
from jedi.api.project import Project
|
|
script = Script('''from datetime import datetime
|
|
d1 = datetime.now()
|
|
d2 = datetime.now()
|
|
''', project=Project('', sys_path=[], smart_sys_path=False))
|
|
now_places = script.get_references(2, 14, scope='file')
|
|
assert len(now_places) == 2
|
|
assert now_places == script.get_references(3, 14, scope='file')
|
|
|
|
|
|
def test_local_references_kwarg(Script):
|
|
from jedi.api.project import Project
|
|
script = Script('''from jedi import Script
|
|
Script(code='')
|
|
''', project=Project('', sys_path=[], smart_sys_path=False))
|
|
assert len(script.get_references(2, 7, scope='file')) == 1
|