incomplete functions shouldn't cause any trouble. fixes #429.

jedi/parser/__init__.py

@@ -555,8 +555,9 @@ class Parser(object):
             elif tok_str in ('return', 'yield'):
                 s = tok.start_pos
                 self.freshscope = False
-                # add returns to the scope
-                func = self._scope.get_parent_until(pr.Function)
+                # Add returns to the scope
+                # Should be a function, otherwise just add it to a module!
+                func = self._scope.get_parent_until((pr.Function, pr.Module))
                 if tok_str == 'yield':
                     func.is_generator = True
 

test/test_parser/test_fast_parser.py

@@ -78,3 +78,10 @@ def test_change_and_undo():
     fp('a')
     fp('b')
     fp('a')
+
+
+def test_incomplete_function():
+    source = '''return ImportErr'''
+
+    script = jedi.Script(dedent(source), 1, 3)
+    assert script.completions()