Make generators return more correct values with while loops, fixes #683

2020-01-29 10:13:46 +01:00
parent d630ed55f3
commit e930f47861
4 changed files with 13 additions and 8 deletions
--- a/jedi/inference/value/function.py
+++ b/jedi/inference/value/function.py
@@ -265,7 +265,7 @@ class BaseFunctionExecutionContext(ValueContext, TreeContextMixin):
            else:
                types = self.get_return_values(check_yields=True)
                if types:
-                    yield LazyKnownValues(types)
+                    yield LazyKnownValues(types, min=0, max=float('inf'))
                return
            last_for_stmt = for_stmt