Merge pull request #1975 from flaviut/fix-unittest

Fix named argument handling

1 files changed, 11 insertions, 2 deletions

diff --git a/lib/pure/unittest.nim b/lib/pure/unittest.nim
index d42941420..8dc9fe0d4 100644
--- a/lib/pure/unittest.nim
+++ b/lib/pure/unittest.nim
@@ -149,10 +149,19 @@ macro check*(conditions: stmt): stmt {.immediate.} =
         inc counter
         var arg = newIdentNode(":p" & $counter)
         var argStr = exp[i].toStrLit
+        var paramAst = exp[i]
         if exp[i].kind in nnkCallKinds: inspectArgs(exp[i])
-        argsAsgns.add getAst(asgn(arg, exp[i]))
+        if exp[i].kind == nnkExprEqExpr:
+          # ExprEqExpr
+          #   Ident !"v"
+          #   IntLit 2
+          paramAst = exp[i][1]
+        argsAsgns.add getAst(asgn(arg, paramAst))
         argsPrintOuts.add getAst(print(argStr, arg))
-        exp[i] = arg
+        if exp[i].kind != nnkExprEqExpr:
+          exp[i] = arg
+        else:
+          exp[i][1] = arg
 
   case checked.kind
   of nnkCallKinds: