diff --git a/test/test_api_classes.py b/test/test_api_classes.py new file mode 100644 index 00000000..3856d6be --- /dev/null +++ b/test/test_api_classes.py @@ -0,0 +1,23 @@ +import pytest + +from jedi import api + + +def make_definitions(): + return api.defined_names(""" + import sys + + class C: + pass + + x = C() + + def f(): + pass + """) + + +@pytest.mark.parametrize('definition', make_definitions()) +def test_basedefinition_type(definition): + assert definition.type in ('module', 'class', 'instance', 'function', + 'statement', 'import')