Add some tests

Author: YingboMa

test/build_function.jl

@@ -2,23 +2,33 @@ using ModelingToolkit, Test
 @variables a b c1 c2 c3 d e g
 
 # Multiple argument matrix
-h = [a + b + c1 + c2; c3 + d + e + g] # uses the same number of arguments as our application
-h_julia(a, b, c, d, e, g) = [a[1] + b[1] + c[1] + c[2]; c[3] + d[1] + e[1] + g[1]]
+h = [a + b + c1 + c2,
+     c3 + d + e + g,
+     0] # uses the same number of arguments as our application
+h_julia(a, b, c, d, e, g) = [a[1] + b[1] + c[1] + c[2],
+                             c[3] + d[1] + e[1] + g[1],
+                             0]
 function h_julia!(out, a, b, c, d, e, g)
-    out .= [a[1] + b[1] + c[1] + c[2]; c[3] + d[1] + e[1] + g[1]]
+    out .= [a[1] + b[1] + c[1] + c[2], c[3] + d[1] + e[1] + g[1], 0]
 end
 
 h_str = ModelingToolkit.build_function(h, [a], [b], [c1, c2, c3], [d], [e], [g])
 h_oop = eval(h_str[1])
 h_ip! = eval(h_str[2])
+h_ip_skip! = eval(ModelingToolkit.build_function(h, [a], [b], [c1, c2, c3], [d], [e], [g], skipzeros=true)[2])
 inputs = ([1], [2], [3, 4, 5], [6], [7], [8])
 
 @test h_oop(inputs...) == h_julia(inputs...)
-out_1 = Array{Int64}(undef, 2)
+out_1 = similar(h, Int)
 out_2 = similar(out_1)
 h_ip!(out_1, inputs...)
 h_julia!(out_2, inputs...)
 @test out_1 == out_2
+fill!(out_1, 10)
+h_ip_skip!(out_1, inputs...)
+@test out_1[3] == 10
+out_1[3] = 0
+@test out_1 == out_2
 
 # Multiple input matrix, some unused arguments
 h_skip = [a + b + c1; c2 + c3 + g] # skip d, e

test/build_function_arrayofarray.jl

@@ -18,11 +18,16 @@ function h_dense_arraymat_julia!(out, x)
     out[3] .= [a[1] c[1]; 1 0]
 end
 
-h_dense_arraymat_str = ModelingToolkit.build_function(h_dense_arraymat, [a, b, c])
-h_dense_arraymat_ip! = eval(h_dense_arraymat_str[2])
-out_1_arraymat = [Array{Int64}(undef, 2, 2) for i in 1:3]
-out_2_arraymat = [similar(x) for x in out_1_arraymat]
+h_dense_arraymat_ip! = eval(ModelingToolkit.build_function(h_dense_arraymat, [a, b, c])[2])
+h_dense_arraymat_ip_skip! = eval(ModelingToolkit.build_function(h_dense_arraymat, [a, b, c], skipzeros=true)[2])
+out_1_arraymat = [fill(42, 2, 2) for i in 1:3]
+out_2_arraymat = deepcopy(out_1_arraymat)
 h_dense_arraymat_julia!(out_1_arraymat, input)
+h_dense_arraymat_ip_skip!(out_2_arraymat, input)
+@test all(isequal(42), out_2_arraymat[2])
+foreach(mat->fill!(mat, 0), out_2_arraymat)
+h_dense_arraymat_ip_skip!(out_2_arraymat, input)
+@test out_1_arraymat == out_2_arraymat
 h_dense_arraymat_ip!(out_2_arraymat, input)
 @test out_1_arraymat == out_2_arraymat