From 8e27ed556e5b0bb2cb86770a051ed96858b59c67 Mon Sep 17 00:00:00 2001
From: Dave Halter <davidhalter88@gmail.com>
Date: Fri, 16 May 2014 12:33:00 +0200
Subject: [PATCH] fix returns/flow command edge cases

---
 jedi/evaluate/analysis.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/jedi/evaluate/analysis.py b/jedi/evaluate/analysis.py
index 152b2f4c..de1e93a1 100644
--- a/jedi/evaluate/analysis.py
+++ b/jedi/evaluate/analysis.py
@@ -114,6 +114,7 @@ def get_module_statements(module):
         new = set()
         for stmt in stmts:
             if isinstance(stmt, pr.Scope):
+                new |= add_stmts(stmt.inputs)
                 continue
             if isinstance(stmt, pr.KeywordStatement):
                 stmt = stmt.stmt
@@ -136,4 +137,5 @@ def get_module_statements(module):
     for scope in module.walk():
         imports |= set(scope.imports)
         stmts |= add_stmts(scope.statements)
+        stmts |= add_stmts(scope.returns)
     return stmts, imports