|
| 1 | +@testitem "Initial guess providing" begin |
| 2 | + using BoundaryValueDiffEq |
| 3 | + tspan = (0.0, pi / 2) |
| 4 | + function simplependulum!(du, u, p, t) |
| 5 | + θ = u[1] |
| 6 | + dθ = u[2] |
| 7 | + du[1] = dθ |
| 8 | + du[2] = -9.81 * sin(θ) |
| 9 | + end |
| 10 | + function bc!(residual, u, p, t) |
| 11 | + residual[1] = u(pi / 4)[1] + pi / 2 |
| 12 | + residual[2] = u(pi / 2)[1] - pi / 2 |
| 13 | + end |
| 14 | + u0 = [pi / 2, pi / 2] |
| 15 | + prob = BVProblem(simplependulum!, bc!, u0, tspan) |
| 16 | + sol1 = solve(prob, MIRK4(), dt = 0.05) |
| 17 | + |
| 18 | + # Solution |
| 19 | + prob1 = BVProblem(simplependulum!, bc!, sol1, tspan) |
| 20 | + sol2 = solve(prob1, MIRK4()) |
| 21 | + @test SciMLBase.successful_retcode(sol2) |
| 22 | + |
| 23 | + sol3 = solve(prob1, RadauIIa5()) |
| 24 | + @test SciMLBase.successful_retcode(sol3) |
| 25 | + |
| 26 | + sol4 = solve(prob1, LobattoIIIa4(nested_nlsolve = true)) |
| 27 | + @test SciMLBase.successful_retcode(sol4) |
| 28 | + |
| 29 | + # VectorOfArray |
| 30 | + prob2 = BVProblem(simplependulum!, bc!, VectorOfArray(sol1.u), tspan) |
| 31 | + sol2 = solve(prob2, MIRK4()) |
| 32 | + @test SciMLBase.successful_retcode(sol2) |
| 33 | + |
| 34 | + sol3 = solve(prob2, RadauIIa5()) |
| 35 | + @test SciMLBase.successful_retcode(sol3) |
| 36 | + |
| 37 | + sol4 = solve(prob2, LobattoIIIa4(nested_nlsolve = true)) |
| 38 | + @test SciMLBase.successful_retcode(sol4) |
| 39 | +end |
0 commit comments