fix cases from issue #120.

4 files changed, 11 insertions, 3 deletions

diff --git a/spec/inputs/loops.yue b/spec/inputs/loops.yue
index d997c65..51eb10b 100644
--- a/spec/inputs/loops.yue
+++ b/spec/inputs/loops.yue
@@ -75,15 +75,17 @@ while also do
 i = 0
 x = while i < 10
         i += 1
+        i
 
--- values that can'e be coerced
+-- values that can't be coerced
 
 x = for thing in *3
         y = "hello"
+        break
 
 x = for x=1,2
         y = "hello"
-
+        y
 
 -- continue
 
diff --git a/spec/inputs/return.yue b/spec/inputs/return.yue
index 96fa0cd..fda8d62 100644
--- a/spec/inputs/return.yue
+++ b/spec/inputs/return.yue
@@ -6,7 +6,7 @@ _ = -> [x for x in *things]
 
 -- doesn't make sense on purpose
 do
-        return x for x in *things
+        for x in *things do return x
 
 do
         return [x for x in *things]
diff --git a/spec/outputs/5.1/loops.lua b/spec/outputs/5.1/loops.lua
index a3b9548..85e9de8 100644
--- a/spec/outputs/5.1/loops.lua
+++ b/spec/outputs/5.1/loops.lua
@@ -125,6 +125,7 @@ do
         local _len_0 = 1
         while i < 10 do
                 i = i + 1
+                _accum_0[_len_0] = i
                 _len_0 = _len_0 + 1
         end
         x = _accum_0
@@ -136,6 +137,7 @@ do
         for _index_0 = 1, #_list_2 do
                 local thing = _list_2[_index_0]
                 y = "hello"
+                break
                 _len_0 = _len_0 + 1
         end
         x = _accum_0
@@ -145,6 +147,7 @@ do
         local _len_0 = 1
         for x = 1, 2 do
                 y = "hello"
+                _accum_0[_len_0] = y
                 _len_0 = _len_0 + 1
         end
         x = _accum_0
diff --git a/spec/outputs/loops.lua b/spec/outputs/loops.lua
index 9c16ee4..eeea15f 100644
--- a/spec/outputs/loops.lua
+++ b/spec/outputs/loops.lua
@@ -125,6 +125,7 @@ do
         local _len_0 = 1
         while i < 10 do
                 i = i + 1
+                _accum_0[_len_0] = i
                 _len_0 = _len_0 + 1
         end
         x = _accum_0
@@ -136,6 +137,7 @@ do
         for _index_0 = 1, #_list_2 do
                 local thing = _list_2[_index_0]
                 y = "hello"
+                break
                 _len_0 = _len_0 + 1
         end
         x = _accum_0
@@ -145,6 +147,7 @@ do
         local _len_0 = 1
         for x = 1, 2 do
                 y = "hello"
+                _accum_0[_len_0] = y
                 _len_0 = _len_0 + 1
         end
         x = _accum_0