orbital multiple shooting

SciML · Sep 23, 2023 · 95d6205 · 95d6205
1 parent aafc2bb
commit 95d6205
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/test/orbital.jl b/test/orbital.jl
@@ -71,10 +71,17 @@ for autodiff in (AutoForwardDiff(), AutoFiniteDiff(; fdtype = Val(:central)),
     AutoSparseForwardDiff(), AutoFiniteDiff(; fdtype = Val(:forward)),
     AutoSparseFiniteDiff())
     nlsolve = NewtonRaphson(; autodiff)
-    @time sol = solve(bvp, Shooting(DP5(); nlsolve); force_dtmin = true, abstol = 1e-13,
-        reltol = 1e-13)
+
+    @time sol = solve(bvp, Shooting(DP5(); nlsolve); force_dtmin = true,
+        abstol = 1e-13, reltol = 1e-13)
     cur_bc!(resid_f, sol, nothing, sol.t)
     @test norm(resid_f, Inf) < TestTol
+
+    @time sol = solve(bvp, MultipleShooting(10, DP5(); nlsolve); abstol = 1e-6,
+        reltol = 1e-6)
+    @test SciMLBase.successful_retcode(sol)
+    cur_bc!(resid_f, sol, nothing, sol.t)
+    @test norm(resid_f, Inf) < 1e-6
 end
 
 ### Using the TwoPoint BVP Structure
@@ -84,8 +91,15 @@ for autodiff in (AutoForwardDiff(), AutoFiniteDiff(; fdtype = Val(:central)),
     AutoSparseForwardDiff(), AutoFiniteDiff(; fdtype = Val(:forward)),
     AutoSparseFiniteDiff())
     nlsolve = NewtonRaphson(; autodiff)
-    @time sol = solve(bvp, Shooting(DP5(); nlsolve); force_dtmin = true, abstol = 1e-13,
-        reltol = 1e-13)
+
+    @time sol = solve(bvp, Shooting(DP5(); nlsolve); force_dtmin = true,
+        abstol = 1e-13, reltol = 1e-13)
+    cur_bc_2point!(resid_f_2p, (sol(t0), sol(t1)), nothing)
+    @test norm(vcat(resid_f_2p...), Inf) < TestTol
+
+    @time sol = solve(bvp, MultipleShooting(10, DP5(); nlsolve); abstol = 1e-6,
+        reltol = 1e-6)
+    @test SciMLBase.successful_retcode(sol)
     cur_bc_2point!(resid_f_2p, (sol(t0), sol(t1)), nothing)
     @test norm(vcat(resid_f_2p...), Inf) < TestTol
 end