Fix a few more tests

2017-11-26 22:07:13 +01:00
parent 85ce57a863
commit accf20226d
6 changed files with 26 additions and 27 deletions
--- a/jedi/evaluate/context/iterable.py
+++ b/jedi/evaluate/context/iterable.py
@@ -250,7 +250,7 @@ class DictComprehension(ArrayMixin, Comprehension):
        for keys, values in self._iterate():
            for k in keys:
                if isinstance(k, compiled.CompiledObject):
-                    if k.obj == index:
+                    if k.get_safe_value(default=object()) == index:
                        return values
        return self.dict_values()