Dict comprehensions are working partially.

2015-12-27 17:20:49 +01:00
parent b3f7d0c29a
commit 03eaf8455f
4 changed files with 25 additions and 8 deletions
--- a/jedi/evaluate/iterable.py
+++ b/jedi/evaluate/iterable.py
@@ -201,6 +201,13 @@ class SetComprehension(Comprehension, ArrayMixin):
    type = 'set'


+class DictComprehension(Comprehension, ArrayMixin):
+    type = 'dict'
+
+    def _get_comp_for(self):
+        return self._get_comprehension().children[3]
+
+
 class GeneratorComprehension(Comprehension, GeneratorMixin):
    pass