Fixed list comprehension name lookups.

2014-12-01 00:08:27 +01:00
parent 3928f466cf
commit 68bd9160e2
4 changed files with 10 additions and 11 deletions
--- a/jedi/evaluate/iterable.py
+++ b/jedi/evaluate/iterable.py
@@ -151,6 +151,9 @@ class Comprehension(IterableWrapper):

        return helpers.deep_ast_copy(comprehension.children[0], {comprehension: last_comp})

+    def get_exact_index_types(self, index):
+        return [self._evaluator.eval_element(self.eval_node())[index]]
+
    def __repr__(self):
        return "<e%s of %s>" % (type(self).__name__, self._atom)