Fix typo and test

Author: YingboMa

src/code.jl

@@ -111,12 +111,12 @@ const NaNMathFuns = (
     :log1p,
     :sqrt,
 )
-function function_to_expr(op, args, st)
-    (op isa Function && (name = nameof(op)) in NaNMathFuns) && return nothing
+function function_to_expr(op, O, st)
+    (op isa Function && (name = nameof(op)) in NaNMathFuns) || return nothing
     fun = GlobalRef(NaNMath, name)
     args = map(Base.Fix2(toexpr, st), arguments(O))
     expr = Expr(:call, fun)
-    expr.args = args
+    append!(expr.args, args)
     return expr
 end
 

test/code.jl

@@ -189,6 +189,10 @@ test_repr(a, b) = @test repr(Base.remove_linenums!(a)) == repr(Base.remove_linen
         f = eval(toexpr(Func([a+b], [], a+b)))
         @test f(1) == 1
         @test f(2) == 2
+
+        f = eval(toexpr(Func([a, b], [], sqrt(a - b))))
+        @test isnan(f(0, 10))
+        @test f(10, 2) ≈ sqrt(8)
     end
 
     let