Add er.Generator.__getattr__

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)
 

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')