Skip to content

Commit ddce985

Browse files
test: use new codegen in sdesystem test
1 parent f9069b1 commit ddce985

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

test/sdesystem.jl

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,14 +24,14 @@ noiseeqs = [0.1 * x,
2424
@named de = SDESystem(eqs, noiseeqs, tt, [x, y, z], [σ, ρ, β], tspan = (0.0, 10.0))
2525
de = complete(de)
2626
f = eval(generate_diffusion_function(de)[1])
27-
@test f(ones(3), rand(3), nothing) == 0.1ones(3)
27+
@test f(ones(3), (rand(3),), nothing) == 0.1ones(3)
2828

2929
f = SDEFunction(de)
30-
prob = SDEProblem(SDEFunction(de), [1.0, 0.0, 0.0], (0.0, 100.0), (10.0, 26.0, 2.33))
30+
prob = SDEProblem(de, [1.0, 0.0, 0.0], (0.0, 100.0), [10.0, 26.0, 2.33])
3131
sol = solve(prob, SRIW1(), seed = 1)
3232

33-
probexpr = SDEProblem(SDEFunction(de), [1.0, 0.0, 0.0], (0.0, 100.0),
34-
(10.0, 26.0, 2.33))
33+
probexpr = SDEProblem(de, [1.0, 0.0, 0.0], (0.0, 100.0),
34+
[10.0, 26.0, 2.33])
3535
solexpr = solve(eval(probexpr), SRIW1(), seed = 1)
3636

3737
@test all(x -> x == 0, Array(sol - solexpr))
@@ -43,13 +43,13 @@ noiseeqs_nd = [0.01*x 0.01*x*y 0.02*x*z
4343
de = complete(de)
4444
f = eval(generate_diffusion_function(de)[1])
4545
p = MTKParameters(de, [σ => 0.1, ρ => 0.2, β => 0.3])
46-
@test f([1, 2, 3.0], p..., nothing) == [0.01*1 0.01*1*2 0.02*1*3
46+
@test f([1, 2, 3.0], p, nothing) == [0.01*1 0.01*1*2 0.02*1*3
4747
0.1 0.01*2 0.02*1*3
4848
0.2 0.3 0.01*3]
4949

5050
f = eval(generate_diffusion_function(de)[2])
5151
du = ones(3, 3)
52-
f(du, [1, 2, 3.0], p..., nothing)
52+
f(du, [1, 2, 3.0], p, nothing)
5353
@test du == [0.01*1 0.01*1*2 0.02*1*3
5454
0.1 0.01*2 0.02*1*3
5555
0.2 0.3 0.01*3]
@@ -86,7 +86,7 @@ function test_SDEFunction_no_eval()
8686
# Need to test within a function scope to trigger world age issues
8787
f = SDEFunction(de, eval_expression = false)
8888
p = MTKParameters(de, [σ => 10.0, ρ => 26.0, β => 2.33])
89-
@test f([1.0, 0.0, 0.0], p..., (0.0, 100.0)) [-10.0, 26.0, 0.0]
89+
@test f([1.0, 0.0, 0.0], p, (0.0, 100.0)) [-10.0, 26.0, 0.0]
9090
end
9191
test_SDEFunction_no_eval()
9292

0 commit comments

Comments
 (0)