Fix python 2 string iterators.

2016-12-03 02:08:40 +01:00
parent ba8a3215f2
commit da1a163da7
4 changed files with 21 additions and 2 deletions
--- a/jedi/evaluate/iterable.py
+++ b/jedi/evaluate/iterable.py
@@ -237,7 +237,8 @@ class Comprehension(AbstractSequence):

    @memoize_default()
    def _get_comp_for_context(self, parent_context, comp_for):
-        return parent_context.create_context(comp_for)
+        # TODO shouldn't this be part of create_context?
+        return CompForContext.from_comp_for(parent_context, comp_for)

    def _nested(self, comp_fors, parent_context=None):
        evaluator = self.evaluator