Fix tests

Author: YingboMa

src/build_function.jl

@@ -585,7 +585,7 @@ function _build_function(target::StanTarget, eqs::Array{<:Equation}, vs, ps, iv;
                                    rhsnames=rhsnames) for
                                    (i, eq) ∈ enumerate(eqs)],";\n  "),";")
     """
-    real[] $fname(real $iv,real[] $(rhsnames[1]),real[] $(rhsnames[2]),real[] x_r,int[] x_i) {
+    real[] $fname(real $(conv(iv)),real[] $(rhsnames[1]),real[] $(rhsnames[2]),real[] x_r,int[] x_i) {
       real $lhsname[$(length(eqs))];
       $differential_equation
       return $lhsname;

src/systems/diffeqs/modelingtoolkitize.jl

@@ -102,9 +102,9 @@ function modelingtoolkitize(prob::DiffEqBase.OptimizationProblem)
         p = prob.p
     end
 
-    vars = reshape([Variable(:x, i)() for i in eachindex(prob.u0)],size(prob.u0))
+    vars = reshape([Variable(:x, i) for i in eachindex(prob.u0)],size(prob.u0))
     params = p isa DiffEqBase.NullParameters ? [] :
-             reshape([Variable(:α,i)() for i in eachindex(p)],size(Array(p)))
+             reshape([Variable(:α,i) for i in eachindex(p)],size(Array(p)))
 
 
     eqs = prob.f(vars, params)

test/variable_utils.jl

@@ -2,7 +2,7 @@ using ModelingToolkit, Test
 @parameters α β δ
 expr = (((1 / β - 1) + δ) / α) ^ (1 / (α - 1))
 ref = [β, δ, α]
-sol = ModelingToolkit.get_variables(expr)
+sol = Num.(ModelingToolkit.get_variables(expr))
 @test all(simplify, sol[i] == ref[i] for i in 1:3)
 
 @parameters γ