@testset "Find Closest" begin using NLsolve using Thesis: treat_inputs # Initial Setup sc = Sc("test") a = rand(15000:1.:40000) e = rand(0.01:0.01:0.5) i = rand(0.01:0.01:π/6) T = 2π*√(a^3/μs["Earth"]) prop_time = 2T n = 50 # A simple orbit raising start = oe_to_xyz([ a, e, i, 0., 0., 0. ], μs["Earth"]) ΔVs = repeat([0.6, 0., 0.]', outer=(n,1)) final = prop(ΔVs, start, sc, μs["Earth"], prop_time)[1][end,:] new_T = 2π*√(xyz_to_oe(final, μs["Earth"])[1]^3/μs["Earth"]) # This should be close enough to 0.6 x0 = repeat([0.59, 0., 0.], n) result = Thesis.single_shoot(start, final, sc, μs["Earth"], 0.0, prop_time, x0) # Test and plot @test converged(result) path1 = prop(zeros((100,3)), start, sc, μs["Earth"], T)[1] path2, mass = prop(treat_inputs(result.zero), start, sc, μs["Earth"], prop_time) path3 = prop(zeros((100,3)), path2[end,:], sc, μs["Earth"], new_T)[1] path4 = prop(zeros((100,3)), final, sc, μs["Earth"], new_T)[1] savefig(plot_orbits([path1, path2, path3, path4], labels=["inital", "transit", "after transit", "final"], colors=["#FFFFFF","#FF4444","#44FF44","#4444FF"]), "../plots/find_closest_test.html") if converged(result) @test norm(path2[end,:] - final) < 1e-4 end end