fix test error

Author: isaacsas

src/systems/diffeqs/abstractodesystem.jl

@@ -215,12 +215,12 @@ function ODEFunctionExpr{iip}(sys::AbstractODESystem, dvs = states(sys),
 
     jp_expr = sparse ? :(similar($(sys.jac[]),Float64)) : :nothing
 
+    sts = states(sys)
     ex = quote
         f = $f
         tgrad = $_tgrad
         jac = $_jac
         M = $_M
-        sts = $(states(sys))
         ODEFunction{$iip}(f,
                          jac = jac,
                          tgrad = tgrad,

src/systems/diffeqs/sdesystem.jl

@@ -255,6 +255,7 @@ function SDEFunctionExpr{iip}(sys::SDESystem, dvs = states(sys),
 
     _M = (u0 === nothing || M == I) ? M : ArrayInterface.restructure(u0 .* u0',M)
 
+    sts = states(sys)
     ex = quote
         f = $f
         g = $g
@@ -263,7 +264,6 @@ function SDEFunctionExpr{iip}(sys::SDESystem, dvs = states(sys),
         Wfact = $_Wfact
         Wfact_t = $_Wfact_t
         M = $_M
-        sts = $(states(sys))
         SDEFunction{$iip}(f,g,
                          jac = jac,
                          tgrad = tgrad,

src/systems/jumps/jumpsystem.jl

@@ -226,12 +226,12 @@ function DiscreteProblemExpr(sys::JumpSystem, u0map, tspan::Tuple,
     u0 = varmap_to_vars(u0map, states(sys))
     p  = varmap_to_vars(parammap, parameters(sys))
     # identity function to make syms works
+    sts = states(sys)
     quote
         f  = DiffEqBase.DISCRETE_INPLACE_DEFAULT
         u0 = $u0
         p = $p
         tspan = $tspan
-        sts = $(states(sys))
         df = DiscreteFunction{true,true}(f, syms=$(tosymbol.(sts, states=sts, escape=false)))
         DiscreteProblem(df, u0, tspan, p; kwargs...)
     end