From 6a2e535bcbb20d8331ba7bab749c03725aea7f25 Mon Sep 17 00:00:00 2001 From: Takafumi Arakaki Date: Sat, 16 Mar 2013 21:49:46 +0100 Subject: [PATCH] Add er.Generator.__getattr__ --- jedi/evaluate_representation.py | 3 +++ test/test_api_classes.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) 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')