13 files changed, 232 insertions, 1 deletions

diff --git a/spec/outputs/comprehension_nested.lua b/spec/outputs/comprehension_nested.lua
new file mode 100644
index 0000000..abd5792
--- /dev/null
+++ b/spec/outputs/comprehension_nested.lua
@@ -0,0 +1,24 @@
+local list = {
+        1,
+        2,
+        3
+}
+local pairs
+do
+        local _accum_0 = { }
+        local _len_0 = 1
+        for _index_0 = 1, #list do
+                local i = list[_index_0]
+                if i % 2 == 1 then
+                        for _index_1 = 1, #list do
+                                local j = list[_index_1]
+                                if j > i then
+                                        _accum_0[_len_0] = tostring(i) .. "-" .. tostring(j)
+                                        _len_0 = _len_0 + 1
+                                end
+                        end
+                end
+        end
+        pairs = _accum_0
+end
+return print(table.concat(pairs, ","))
diff --git a/spec/outputs/destructure_defaults.lua b/spec/outputs/destructure_defaults.lua
new file mode 100644
index 0000000..938d9b2
--- /dev/null
+++ b/spec/outputs/destructure_defaults.lua
@@ -0,0 +1,15 @@
+local t = {
+        a = 1,
+        b = {
+                c = 3
+        },
+        d = nil
+}
+local a, c, d, e = t.a, t.b.c, t.b.d, t.e
+if d == nil then
+        d = 4
+end
+if e == nil then
+        e = 5
+end
+return print(a, c, d, e)
diff --git a/spec/outputs/export_mixed.lua b/spec/outputs/export_mixed.lua
new file mode 100644
index 0000000..35529e2
--- /dev/null
+++ b/spec/outputs/export_mixed.lua
@@ -0,0 +1,11 @@
+local _module_0 = { }
+local answer = 42
+_module_0["answer"] = answer
+local foo
+foo = function()
+        return "bar"
+end
+_module_0["foo"] = foo
+local baz = "baz"
+_module_0["baz"] = baz
+return _module_0
diff --git a/spec/outputs/nil_coalesce_precedence.lua b/spec/outputs/nil_coalesce_precedence.lua
new file mode 100644
index 0000000..6f353ee
--- /dev/null
+++ b/spec/outputs/nil_coalesce_precedence.lua
@@ -0,0 +1,24 @@
+local a = nil
+local b = false
+local c = 0
+local x
+if a ~= nil then
+        x = a
+else
+        do
+                local _exp_0 = (b and 1)
+                if _exp_0 ~= nil then
+                        x = _exp_0
+                else
+                        x = (c or 2)
+                end
+        end
+end
+local y = ((function()
+        if a ~= nil then
+                return a
+        else
+                return b
+        end
+end)()) and 1 or 2
+return print(x, y)
diff --git a/spec/outputs/pipe_chain_combo.lua b/spec/outputs/pipe_chain_combo.lua
new file mode 100644
index 0000000..6cf40a9
--- /dev/null
+++ b/spec/outputs/pipe_chain_combo.lua
@@ -0,0 +1,9 @@
+local f1
+f1 = function(x)
+        return x + 2
+end
+local f2
+f2 = function(x)
+        return x * 3
+end
+local value = print(tostring(f2(f1(3))))
diff --git a/spec/outputs/test/comprehension_spec.lua b/spec/outputs/test/comprehension_spec.lua
new file mode 100644
index 0000000..a4682d5
--- /dev/null
+++ b/spec/outputs/test/comprehension_spec.lua
@@ -0,0 +1,31 @@
+return describe("comprehension", function()
+        return it("nested with filter", function()
+                local list = {
+                        1,
+                        2,
+                        3
+                }
+                local out
+                do
+                        local _accum_0 = { }
+                        local _len_0 = 1
+                        for _index_0 = 1, #list do
+                                local i = list[_index_0]
+                                if i % 2 == 1 then
+                                        for _index_1 = 1, #list do
+                                                local j = list[_index_1]
+                                                if j > i then
+                                                        _accum_0[_len_0] = tostring(i) .. "-" .. tostring(j)
+                                                        _len_0 = _len_0 + 1
+                                                end
+                                        end
+                                end
+                        end
+                        out = _accum_0
+                end
+                return assert.same(out, {
+                        "1-2",
+                        "1-3"
+                })
+        end)
+end)
diff --git a/spec/outputs/test/destructure_spec.lua b/spec/outputs/test/destructure_spec.lua
new file mode 100644
index 0000000..4e18b02
--- /dev/null
+++ b/spec/outputs/test/destructure_spec.lua
@@ -0,0 +1,29 @@
+return describe("destructure", function()
+        return it("defaults and nested", function()
+                local t = {
+                        a = 1,
+                        b = {
+                                c = 3
+                        },
+                        d = nil
+                }
+                local a, c, d, e = t.a, t.b.c, t.b.d, t.e
+                if d == nil then
+                        d = 4
+                end
+                if e == nil then
+                        e = 5
+                end
+                return assert.same({
+                        a,
+                        c,
+                        d,
+                        e
+                }, {
+                        1,
+                        3,
+                        4,
+                        5
+                })
+        end)
+end)
diff --git a/spec/outputs/test/format_spec.lua b/spec/outputs/test/format_spec.lua
index 898fa03..74a511d 100644
--- a/spec/outputs/test/format_spec.lua
+++ b/spec/outputs/test/format_spec.lua
@@ -94,7 +94,19 @@ local files = {
         "spec/inputs/unicode/metatable.yue",
         "spec/inputs/unicode/syntax.yue",
         "spec/inputs/unicode/global.yue",
-        "spec/inputs/unicode/plus.yue"
+        "spec/inputs/unicode/plus.yue",
+        "spec/inputs/pipe_chain_combo.yue",
+        "spec/inputs/destructure_defaults.yue",
+        "spec/inputs/nil_coalesce_precedence.yue",
+        "spec/inputs/comprehension_nested.yue",
+        "spec/inputs/with_scope_shadow.yue",
+        "spec/inputs/export_mixed.yue",
+        "spec/inputs/unicode/pipe_chain_combo.yue",
+        "spec/inputs/test/destructure_spec.yue",
+        "spec/inputs/test/nil_coalescing_spec.yue",
+        "spec/inputs/test/pipe_spec.yue",
+        "spec/inputs/test/try_catch_spec.yue",
+        "spec/inputs/test/comprehension_spec.yue"
 }
 local yue = require("yue")
 local rewriteLineCol
diff --git a/spec/outputs/test/nil_coalescing_spec.lua b/spec/outputs/test/nil_coalescing_spec.lua
new file mode 100644
index 0000000..90b2a64
--- /dev/null
+++ b/spec/outputs/test/nil_coalescing_spec.lua
@@ -0,0 +1,23 @@
+local _anon_func_0 = function(a, b)
+        if a ~= nil then
+                return a
+        else
+                return b
+        end
+end
+local _anon_func_1 = function(a, c)
+        if a ~= nil then
+                return a
+        else
+                return c
+        end
+end
+return describe("nil coalescing", function()
+        return it("distinguish nil and false", function()
+                local a = nil
+                local b = false
+                local c = 0
+                assert.same((_anon_func_0(a, b)), false)
+                return assert.same((_anon_func_1(a, c)), 0)
+        end)
+end)
diff --git a/spec/outputs/test/pipe_spec.lua b/spec/outputs/test/pipe_spec.lua
new file mode 100644
index 0000000..3d797ed
--- /dev/null
+++ b/spec/outputs/test/pipe_spec.lua
@@ -0,0 +1,13 @@
+return describe("pipe", function()
+        return it("pipes through functions", function()
+                local f
+                f = function(x)
+                        return x + 1
+                end
+                local g
+                g = function(x)
+                        return x * 2
+                end
+                return assert.same((g(f(3))), 8)
+        end)
+end)
diff --git a/spec/outputs/test/try_catch_spec.lua b/spec/outputs/test/try_catch_spec.lua
new file mode 100644
index 0000000..2bcfc16
--- /dev/null
+++ b/spec/outputs/test/try_catch_spec.lua
@@ -0,0 +1,17 @@
+return describe("try/catch", function()
+        return it("catch and rethrow", function()
+                local ok, success, err = pcall(function()
+                        return xpcall(function()
+                                return error("boom")
+                        end, function(e)
+                                local _, result = pcall(function()
+                                        return error("wrap:" .. e:match("^.-:%d+:%s*(.*)$"))
+                                end)
+                                return result
+                        end)
+                end)
+                assert.same(ok, true)
+                assert.same(success, false)
+                return assert.is_true(err:match("wrap:boom") ~= nil)
+        end)
+end)
diff --git a/spec/outputs/unicode/pipe_chain_combo.lua b/spec/outputs/unicode/pipe_chain_combo.lua
new file mode 100644
index 0000000..f48cb5b
--- /dev/null
+++ b/spec/outputs/unicode/pipe_chain_combo.lua
@@ -0,0 +1,9 @@
+local _u52a0
+_u52a0 = function(x)
+        return x + 2
+end
+local _u4e58
+_u4e58 = function(x)
+        return x * 3
+end
+local _u503c = print(tostring(_u4e58(_u52a0(3))))
diff --git a/spec/outputs/with_scope_shadow.lua b/spec/outputs/with_scope_shadow.lua
new file mode 100644
index 0000000..72abe95
--- /dev/null
+++ b/spec/outputs/with_scope_shadow.lua
@@ -0,0 +1,14 @@
+local target = {
+        val = 1,
+        add = function(self, n)
+                self.val = self.val + n
+                return self.val
+        end
+}
+local result
+do
+        local val = 100
+        add(2)
+        result = target
+end
+return print(result, target.val)