Re-packaged

julia/test/find_closest.jl

@@ -0,0 +1,36 @@
+@testset "Find Closest" begin
+
+  # 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 result[3] == :XTOL_REACHED
+  path1 = prop(zeros((100,3)), start, sc, μs["Earth"], T)[1]
+  path2, mass = prop(treat_inputs(result[2]), 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 result[3] == :XTOL_REACHED
+    @test norm(path2[end,:] - final) < 1e-6
+  end
+
+end