Fix a wrong test about references

This commit is contained in:
Dave Halter
2021-01-02 01:17:38 +01:00
parent 55c7e4eb49
commit 02d43caa5e
2 changed files with 6 additions and 7 deletions

View File

@@ -393,7 +393,7 @@ class Script:
quite hard to do for Jedi, if it is too complicated, Jedi will stop quite hard to do for Jedi, if it is too complicated, Jedi will stop
searching. searching.
:param include_builtins: Default ``True``. If ``False``, checks if a reference :param include_builtins: Default ``True``. If ``False``, checks if a definition
is a builtin (e.g. ``sys``) and in that case does not return it. is a builtin (e.g. ``sys``) and in that case does not return it.
:param scope: Default ``'project'``. If ``'file'``, include references in :param scope: Default ``'project'``. If ``'file'``, include references in
the current module only. the current module only.

View File

@@ -10,16 +10,15 @@ def test_import_references(Script):
def test_exclude_builtin_modules(Script): def test_exclude_builtin_modules(Script):
def get(include): def get(include):
from jedi.api.project import Project references = Script(source).get_references(include_builtins=include)
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] return [(d.line, d.column) for d in references]
source = '''import sys\nprint(sys.path)''' source = '''import sys\nsys.setprofile'''
places = get(include=True) places = get(include=True)
assert len(places) > 2 # Includes stubs assert len(places) >= 3 # Includes stubs, the reference itself and the builtin
places = get(include=False) places = get(include=False)
assert places == [(1, 7), (2, 6)] # Just the reference
assert places == [(2, 4)]
@pytest.mark.parametrize('code, places', [ @pytest.mark.parametrize('code, places', [