Fixing issue #178.

2 files changed, 20 insertions, 0 deletions

diff --git a/spec/inputs/macro.yue b/spec/inputs/macro.yue
index 5dafd84..5d5f1a9 100644
--- a/spec/inputs/macro.yue
+++ b/spec/inputs/macro.yue
@@ -340,6 +340,15 @@ _1 = ->
         print 2
         print 3
 
+do
+        macro foo = -> code: "tb:func(123)", type: "lua"
+        f = ->
+                x = $foo\bar 456
+                $foo!
+        f1 = ->
+                $foo!
+                return
+
 macro implicitReturnMacroIsAllowed = -> "print 'abc'\n123"
 
 $implicitReturnMacroIsAllowed
diff --git a/spec/outputs/macro.lua b/spec/outputs/macro.lua
index 83c6a90..7163077 100644
--- a/spec/outputs/macro.lua
+++ b/spec/outputs/macro.lua
@@ -330,6 +330,17 @@ _1 = function()
         return _accum_0
 end
 do
+        f = function()
+                x = (tb:func(123)):bar(456)
+                return (tb:func(123))
+        end
+        local f1
+        f1 = function()
+tb:func(123)
+                return
+        end
+end
+do
         print('abc')
         return 123
 end