|
1 | 1 | using DiffEqBase, JumpProcesses, OrdinaryDiffEq, StochasticDiffEq, Test |
2 | 2 | using Random, LinearSolve, Statistics |
3 | | -using StableRNGs |
| 3 | +using StableRNGs, ADTypes |
4 | 4 | rng = StableRNG(12345) |
5 | 5 |
|
6 | 6 | a = ExtendedJumpArray(rand(rng, 3), rand(rng, 2)) |
@@ -33,13 +33,13 @@ prob = ODEProblem(f, [0.2], (0.0, 10.0)) |
33 | 33 | jump_prob = JumpProblem(prob, jump, jump2; vr_aggregator = VR_FRM(), rng) |
34 | 34 | integrator = init(jump_prob, Tsit5()) |
35 | 35 | sol = solve(jump_prob, Tsit5()) |
36 | | -sol = solve(jump_prob, Rosenbrock23(autodiff = false)) |
| 36 | +sol = solve(jump_prob, Rosenbrock23(autodiff = AutoFiniteDiff())) |
37 | 37 | sol = solve(jump_prob, Rosenbrock23()) |
38 | 38 |
|
39 | 39 | jump_prob_gill = JumpProblem(prob, jump, jump2; vr_aggregator = VR_Direct(), rng) |
40 | 40 | integrator = init(jump_prob_gill, Tsit5()) |
41 | 41 | sol_gill = solve(jump_prob_gill, Tsit5()) |
42 | | -sol_gill = solve(jump_prob, Rosenbrock23(autodiff = false)) |
| 42 | +sol_gill = solve(jump_prob, Rosenbrock23(autodiff = AutoFiniteDiff())) |
43 | 43 | sol_gill = solve(jump_prob, Rosenbrock23()) |
44 | 44 | @test maximum([sol.u[i][2] for i in 1:length(sol)]) <= 1e-12 |
45 | 45 | @test maximum([sol.u[i][3] for i in 1:length(sol)]) <= 1e-12 |
|
0 commit comments