Merge pull request #1331 from SciML/baggepinnen-patch-1

Propagate state names to DiscreteProblem

Author: ChrisRackauckas

src/systems/discrete_system/discrete_system.jl

@@ -190,7 +190,8 @@ function DiffEqBase.DiscreteProblem(sys::DiscreteSystem,u0map,tspan,
     f_gen = generate_function(sys; expression=Val{eval_expression}, expression_module=eval_module)
     f_oop, _ = (@RuntimeGeneratedFunction(eval_module, ex) for ex in f_gen)
     f(u,p,iv) = f_oop(u,p,iv)
-    DiscreteProblem(f,u0,tspan,p;kwargs...)
+    fd = DiscreteFunction(f, syms=Symbol.(dvs))
+    DiscreteProblem(fd,u0,tspan,p;kwargs...)
 end
 
 function linearize_eqs(sys, eqs=get_eqs(sys); return_max_delay=false)

test/discretesystem.jl

@@ -34,6 +34,7 @@ prob_map = DiscreteProblem(sys,u0,tspan,p)
 # Solution
 using OrdinaryDiffEq
 sol_map = solve(prob_map,FunctionMap());
+@test sol_map[S] isa Vector
 
 # Using defaults constructor
 @parameters t c=10.0 nsteps=400 δt=0.1 β=0.05 γ=0.25