Fix formatting

SciML · Oct 2, 2023 · bc87422 · bc87422
1 parent 1bcc1d9
commit bc87422
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 25 deletions.
diff --git a/test/misc/type_stability.jl b/test/misc/type_stability.jl
@@ -23,29 +23,41 @@ p = [1.0, 1.0, 1.0, 1.0]
 bcresid_prototype = (zeros(1), zeros(1))
 
 # Multi-Point BVP
-mpbvp_iip = BVProblem(f!, bc!, u0, tspan, p)
-mpbvp_oop = BVProblem(f, bc, u0, tspan, p)
-
-@inferred solve(mpbvp_iip, Shooting(Tsit5()))
-@inferred solve(mpbvp_oop, Shooting(Tsit5()))
-@inferred solve(mpbvp_iip, MultipleShooting(5, Tsit5()))
-@inferred solve(mpbvp_oop, MultipleShooting(5, Tsit5()))
-
-for solver in (MIRK2(), MIRK3(), MIRK4(), MIRK5(), MIRK6())
-    @inferred solve(mpbvp_iip, solver; dt = 0.2)
-    @inferred solve(mpbvp_oop, solver; dt = 0.2)
+@testset "Multi-Point BVP" begin
+    mpbvp_iip = BVProblem(f!, bc!, u0, tspan, p)
+    mpbvp_oop = BVProblem(f, bc, u0, tspan, p)
+
+    @testset "Shooting Methods" begin
+        @inferred solve(mpbvp_iip, Shooting(Tsit5()))
+        @inferred solve(mpbvp_oop, Shooting(Tsit5()))
+        @inferred solve(mpbvp_iip, MultipleShooting(5, Tsit5()))
+        @inferred solve(mpbvp_oop, MultipleShooting(5, Tsit5()))
+    end
+
+    @testset "MIRK Methods" begin
+        for solver in (MIRK2(), MIRK3(), MIRK4(), MIRK5(), MIRK6())
+            @inferred solve(mpbvp_iip, solver; dt = 0.2)
+            @inferred solve(mpbvp_oop, solver; dt = 0.2)
+        end
+    end
 end
 
 # Two-Point BVP
-tpbvp_iip = TwoPointBVProblem(f!, twobc!, u0, tspan, p; bcresid_prototype)
-tpbvp_oop = TwoPointBVProblem(f, twobc, u0, tspan, p)
-
-@inferred solve(tpbvp_iip, Shooting(Tsit5()))
-@inferred solve(tpbvp_oop, Shooting(Tsit5()))
-@inferred solve(tpbvp_iip, MultipleShooting(5, Tsit5()))
-@inferred solve(tpbvp_oop, MultipleShooting(5, Tsit5()))
-
-for solver in (MIRK2(), MIRK3(), MIRK4(), MIRK5(), MIRK6())
-    @inferred solve(tpbvp_iip, solver; dt = 0.2)
-    @inferred solve(tpbvp_oop, solver; dt = 0.2)
+@testset "Two-Point BVP" begin
+    tpbvp_iip = TwoPointBVProblem(f!, twobc!, u0, tspan, p; bcresid_prototype)
+    tpbvp_oop = TwoPointBVProblem(f, twobc, u0, tspan, p)
+
+    @testset "Shooting Methods" begin
+        @inferred solve(tpbvp_iip, Shooting(Tsit5()))
+        @inferred solve(tpbvp_oop, Shooting(Tsit5()))
+        @inferred solve(tpbvp_iip, MultipleShooting(5, Tsit5()))
+        @inferred solve(tpbvp_oop, MultipleShooting(5, Tsit5()))
+    end
+
+    @testset "MIRK Methods" begin
+        for solver in (MIRK2(), MIRK3(), MIRK4(), MIRK5(), MIRK6())
+            @inferred solve(tpbvp_iip, solver; dt = 0.2)
+            @inferred solve(tpbvp_oop, solver; dt = 0.2)
+        end
+    end
 end
diff --git a/test/shooting/orbital.jl b/test/shooting/orbital.jl
@@ -73,7 +73,8 @@ 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
 

diff --git a/test/shooting/shooting_tests.jl b/test/shooting/shooting_tests.jl
@@ -90,7 +90,7 @@ end
 
 @testset "Shooting with Complex Values" begin
     # Test for complex values
-        function f1!(du, u, p, t)
+    function f1!(du, u, p, t)
         du[1] = u[2]
         du[2] = -u[1]
         return nothing
@@ -102,6 +102,8 @@ end
         resid[2] = sol(t₁)[1] - 1
         return nothing
     end
+
+    tspan = (0.0, 100.0)
     u0 = [0.0, 1.0] .+ 1im
     bvp = BVProblem(f1!, bc1!, u0, tspan)
     resid_f = Array{ComplexF64}(undef, 2)
@@ -169,7 +171,7 @@ end
 end
 
 @testset "Ray Tracing BVP" begin
-    # Example 1.7 from 
+    # Example 1.7 from
     # "Numerical Solution to Boundary Value Problems for Ordinary Differential equations",
     # 'Ascher, Mattheij, Russell'