fix in-expression format issue.v0.21.7

4 files changed, 91 insertions, 5 deletions

diff --git a/spec/inputs/in_expression.yue b/spec/inputs/in_expression.yue
index 7ee0816..0b2a87a 100644
--- a/spec/inputs/in_expression.yue
+++ b/spec/inputs/in_expression.yue
@@ -1,5 +1,4 @@
 -a^2 in {1, 2, 3} |> f
--a^2 in [1, 2, 3] |> f
 
 has = "foo" in { "bar", "foo" }
 
diff --git a/spec/inputs/unicode/in_expression.yue b/spec/inputs/unicode/in_expression.yue
index e068cbf..4a87099 100644
--- a/spec/inputs/unicode/in_expression.yue
+++ b/spec/inputs/unicode/in_expression.yue
@@ -29,5 +29,44 @@ do
         检查 -> return 对象x in 数组tb
         检查 -> 对象x not in 数组tb
 
+do
+        函数1 = ->
+                变量a = 2
+                变量a in [
+                        1
+                        2
+                        3
+                        4
+                ]
+        函数2 = ->
+                变量a not in [
+                        1
+                        2
+                        3
+                        4
+                ]
+        函数3 = ->
+                变量a = 2
+                变量a in { 变量b, 变量c, 变量d
+                        变量e, 变量f
+                        变量g
+                }
+        函数4 = ->
+                变量a = 2
+                变量b = a in [ 1, 2
+                        3, 4
+                ]
+        函数5 = ->
+                a变量 = 2
+                return a变量 in [ 1, 2
+                        3, 4
+                ]
+        函数6 = ->
+                b变量 = a变量 not in {
+                        1
+                        2
+                        字段x: 3
+                }
+
 nil
 
diff --git a/spec/outputs/in_expression.lua b/spec/outputs/in_expression.lua
index afcc030..b99be8a 100644
--- a/spec/outputs/in_expression.lua
+++ b/spec/outputs/in_expression.lua
@@ -2,10 +2,6 @@ f((function()
         local _val_0 = -a ^ 2
         return 1 == _val_0 or 2 == _val_0 or 3 == _val_0
 end)())
-f((function()
-        local _val_0 = -a ^ 2
-        return 1 == _val_0 or 2 == _val_0 or 3 == _val_0
-end)())
 local has
 do
         local _val_0 = "foo"
diff --git a/spec/outputs/unicode/in_expression.lua b/spec/outputs/unicode/in_expression.lua
index 7c584f3..2f47b1f 100644
--- a/spec/outputs/unicode/in_expression.lua
+++ b/spec/outputs/unicode/in_expression.lua
@@ -153,4 +153,56 @@ do
                 return true
         end)
 end
+do
+        local _u51fd_u65701
+        _u51fd_u65701 = function()
+                local _u53d8_u91cfa = 2
+                return (1 == _u53d8_u91cfa or 2 == _u53d8_u91cfa or 3 == _u53d8_u91cfa or 4 == _u53d8_u91cfa)
+        end
+        local _u51fd_u65702
+        _u51fd_u65702 = function()
+                local _val_0 = _u53d8_u91cfa
+                return not (1 == _val_0 or 2 == _val_0 or 3 == _val_0 or 4 == _val_0)
+        end
+        local _u51fd_u65703
+        _u51fd_u65703 = function()
+                local _u53d8_u91cfa = 2
+                return (_u53d8_u91cfb == _u53d8_u91cfa or _u53d8_u91cfc == _u53d8_u91cfa or _u53d8_u91cfd == _u53d8_u91cfa or _u53d8_u91cfe == _u53d8_u91cfa or _u53d8_u91cff == _u53d8_u91cfa or _u53d8_u91cfg == _u53d8_u91cfa)
+        end
+        local _u51fd_u65704
+        _u51fd_u65704 = function()
+                local _u53d8_u91cfa = 2
+                local _u53d8_u91cfb
+                do
+                        local _val_0 = a
+                        _u53d8_u91cfb = 1 == _val_0 or 2 == _val_0 or 3 == _val_0 or 4 == _val_0
+                end
+        end
+        local _u51fd_u65705
+        _u51fd_u65705 = function()
+                local a_u53d8_u91cf = 2
+                return (1 == a_u53d8_u91cf or 2 == a_u53d8_u91cf or 3 == a_u53d8_u91cf or 4 == a_u53d8_u91cf)
+        end
+        local _u51fd_u65706
+        _u51fd_u65706 = function()
+                local b_u53d8_u91cf
+                do
+                        local _check_0 = {
+                                1,
+                                2,
+                                ["字段x"] = 3
+                        }
+                        local _val_0 = a_u53d8_u91cf
+                        local _find_0 = false
+                        for _index_0 = 1, #_check_0 do
+                                local _item_0 = _check_0[_index_0]
+                                if _item_0 == _val_0 then
+                                        _find_0 = true
+                                        break
+                                end
+                        end
+                        b_u53d8_u91cf = not _find_0
+                end
+        end
+end
 return nil