Adding control input to discretesystem test

Author: cadojo

test/discretesystem.jl

@@ -22,7 +22,7 @@ eqs = [next_S ~ S-infection,
        next_R ~ R+recovery]
 
 # System
-sys = DiscreteSystem(eqs,t,[S,I,R],[c,nsteps,δt,β,γ])
+sys = DiscreteSystem(eqs,t,[S,I,R],[c,nsteps,δt,β,γ]; controls = [β, γ])
 
 # Problem
 u0 = [S => 990.0, I => 10.0, R => 0.0]
@@ -54,4 +54,4 @@ p = [0.05,10.0,0.25,0.1];
 prob_map = DiscreteProblem(sir_map!,u0,tspan,p);
 sol_map2 = solve(prob_map,FunctionMap());
 
-@test Array(sol_map) ≈ Array(sol_map2)
+@test Array(sol_map) ≈ Array(sol_map2)