Skip to content

Commit 388e1b1

Browse files
committed
add tests
1 parent db180fd commit 388e1b1

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

lib/OrdinaryDiffEqStabilizedRK/test/rkc_tests.jl

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,3 +72,33 @@ end
7272
@test sim.𝒪est[:l∞]5 atol=testTol
7373
end
7474
end
75+
76+
@testset "Numer of function evaluations" begin
77+
x = Ref(0)
78+
u0 = [1.0, 1.0]
79+
tspan = (0.0, 1.0)
80+
probop = ODEProblem(u0, tspan) do u, p, t
81+
x[] += 1
82+
return -5 * u
83+
end
84+
probip = ODEProblem(u0, tspan) do du, u, p, t
85+
x[] += 1
86+
@. du = -5 * u
87+
return nothing
88+
end
89+
90+
@testset "$prob" for prob in [probop, probip]
91+
eigen_est = (integrator) -> integrator.eigen_est = 5
92+
algs = [ROCK2(), ROCK2(eigen_est = eigen_est),
93+
ROCK4(), ROCK4(eigen_est = eigen_est),
94+
RKC(), RKC(eigen_est = eigen_est),
95+
SERK2(), SERK2(eigen_est = eigen_est),
96+
ESERK4(), ESERK4(eigen_est = eigen_est),
97+
ESERK5(), ESERK5(eigen_est = eigen_est)]
98+
@testset "$alg" for alg in algs
99+
x[] = 0
100+
sol = solve(prob, alg)
101+
@test x[] == sol.stats.nf
102+
end
103+
end
104+
end

0 commit comments

Comments
 (0)