diff --git a/jedi/evaluate_representation.py b/jedi/evaluate_representation.py index 99fded99..019e038b 100644 --- a/jedi/evaluate_representation.py +++ b/jedi/evaluate_representation.py @@ -773,6 +773,9 @@ class Generator(use_metaclass(cache.CachedMetaClass, pr.Base)): def parent(self): return self.func.parent + def __getattr__(self, name): + return getattr(self.func, name) + def __repr__(self): return "<%s of %s>" % (type(self).__name__, self.func) diff --git a/test/test_api_classes.py b/test/test_api_classes.py index 747eabdc..2da3e13e 100644 --- a/test/test_api_classes.py +++ b/test/test_api_classes.py @@ -50,4 +50,4 @@ def make_definitions(): @pytest.mark.parametrize('definition', make_definitions()) def test_basedefinition_type(definition): assert definition.type in ('module', 'class', 'instance', 'function', - 'statement', 'import', 'param') + 'generator', 'statement', 'import', 'param')