diff --git a/test/test_api_classes.py b/test/test_api_classes.py index 3856d6be..8e4f76cf 100644 --- a/test/test_api_classes.py +++ b/test/test_api_classes.py @@ -1,10 +1,17 @@ +import textwrap + import pytest from jedi import api def make_definitions(): - return api.defined_names(""" + """ + Return a list of definitions for parametrized tests. + + :rtype: [jedi.api_classes.BaseDefinition] + """ + source = textwrap.dedent(""" import sys class C: @@ -16,6 +23,20 @@ def make_definitions(): pass """) + definitions = [] + definitions += api.defined_names(source) + + source += textwrap.dedent(""" + variable = sys or C or x or f""") + lines = source.splitlines() + script = api.Script(source, len(lines), len('variable'), None) + definitions += script.definition() + + script2 = api.Script(source, 4, len('class C'), None) + definitions += script2.related_names() + + return definitions + @pytest.mark.parametrize('definition', make_definitions()) def test_basedefinition_type(definition):