Skip to content

Commit 444d609

Browse files
committed
Add tspan tests
1 parent de23f75 commit 444d609

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

test/discretesystem.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,10 +52,10 @@ eqs2 = [D(S) ~ S - infection2,
5252
D(I) ~ I + infection2 - recovery2,
5353
D(R) ~ R + recovery2]
5454

55-
@named sys = DiscreteSystem(eqs2; controls = [β, γ])
55+
@named sys = DiscreteSystem(eqs2; controls = [β, γ], tspan)
5656
@test ModelingToolkit.defaults(sys) != Dict()
5757

58-
prob_map2 = DiscreteProblem(sys, [], tspan)
58+
prob_map2 = DiscreteProblem(sys)
5959
sol_map2 = solve(prob_map, FunctionMap());
6060

6161
@test sol_map.u == sol_map2.u

test/odesystem.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -374,10 +374,11 @@ eqs = [D(D(x)) ~ -b / M * D(x) - k / M * x]
374374
ps = [M, b, k]
375375
default_u0 = [D(x) => 0.0, x => 10.0]
376376
default_p = [M => 1.0, b => 1.0, k => 1.0]
377-
@named sys = ODESystem(eqs, t, [x], ps, defaults = [default_u0; default_p])
377+
@named sys = ODESystem(eqs, t, [x], ps; defaults = [default_u0; default_p], tspan)
378378
sys = ode_order_lowering(sys)
379-
prob = ODEProblem(sys, [], tspan)
379+
prob = ODEProblem(sys)
380380
sol = solve(prob, Tsit5())
381+
@test sol.t[end] == tspan[end]
381382
@test sum(abs, sol[end]) < 1
382383

383384
# check_eqs_u0 kwarg test

test/sdesystem.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ solexpr = solve(eval(probexpr), SRIW1(), seed = 1)
3636

3737
# Test no error
3838
@test_nowarn SDEProblem(de, nothing, (0, 10.0))
39+
@test_nowarn SDEProblem(de)
3940

4041
noiseeqs_nd = [0.01*x 0.01*x*y 0.02*x*z
4142
σ 0.01*y 0.02*x*z

0 commit comments

Comments
 (0)