Skip to content

Commit

Permalink
🚨 Increase test coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
Ronan Arraes Jardim Chagas authored and Ronan Arraes Jardim Chagas committed Jun 25, 2023
1 parent 3620735 commit fc735b5
Show file tree
Hide file tree
Showing 3 changed files with 540 additions and 146 deletions.
151 changes: 5 additions & 146 deletions test/runtests.jl
View file
Original file line number Diff line number Diff line change
@@ -1,156 +1,15 @@
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
#
# Description
# ==========================================================================================
#
# Test SGP4 algorithm. All tests are based on [1].
#
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
#
# References
# ==========================================================================================
#
# [1] Vallado, D. A., Crawford, P., Hujsak, R., Kelso, T. S (2006). Revisiting Spacetrack
# Report #3: Rev1. AIAA.
#
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #

using Test

using Dates
using DelimitedFiles
using Printf
using SatelliteToolboxTle
using SatelliteToolboxSgp4

@testset "Constructors" begin
sgp4ds = SatelliteToolboxSgp4.Sgp4DeepSpace{Float64}(
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, false, false, false,
)

sgp4c = Sgp4Propagator{Float64, Float64}(
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, :sdp4, sgp4c_wgs84, sgp4ds
)

# Some random tests.
@test sgp4c.epoch == 0.0
@test sgp4c.a_k == 0.0
@test sgp4c.∂M == 0.0
@test sgp4c.algorithm == :sdp4
@test sgp4c.sgp4ds == sgp4ds
@testset "SGP4 Propagator" verbose = true begin
include("./sgp4.jl")
end

@testset "Tests from the Paper AIAA 2006-6753" verbose = true begin
# Read all TLEs that will be used to test.
tles = read_tles_from_file("./sgp4_tests/sgp4_tests.tle")

@testset "Default" begin
for tle in tles
filename = @sprintf(
"./sgp4_tests/aiaa-2006-6753/sgp4_tle_%d_result.txt",
tle.satellite_number
)
SGP4_results = readdlm(filename; comments = true)

# Initialize the orbit propagator.
sgp4d = sgp4_init(tle; sgp4c = sgp4c_wgs72)

t = SGP4_results[:, 1]

@inbounds for k in 1:length(t)

# Propagate the orbit.
r_teme, v_teme = sgp4!(sgp4d, t[k])

# Assemble the result vector.
st_sgp4_result = vcat(t[k], r_teme, v_teme)

# Compare the values.
@test t[k] == SGP4_results[k, 1]
@test r_teme[1] ≈ SGP4_results[k, 2] atol=1e-8
@test r_teme[2] ≈ SGP4_results[k, 3] atol=1e-8
@test r_teme[3] ≈ SGP4_results[k, 4] atol=1e-8
@test v_teme[1] ≈ SGP4_results[k, 5] atol=1e-9
@test v_teme[2] ≈ SGP4_results[k, 6] atol=1e-9
@test v_teme[3] ≈ SGP4_results[k, 7] atol=1e-9
end
end
end

@testset "In-place Initialization" begin
# First, we create a dummy SGP4 structure but with the correct constants and epoch
# type.
sgp4d = sgp4_init(0.0, 0, 0, 0, 0, 0, 0, 0; sgp4c = sgp4c_wgs72)

for tle in tles
filename = @sprintf(
"./sgp4_tests/aiaa-2006-6753/sgp4_tle_%d_result.txt",
tle.satellite_number
)
SGP4_results = readdlm(filename; comments = true)

# Initialize the orbit propagator.
sgp4_init!(sgp4d, tle)

t = SGP4_results[:,1]

@inbounds for k in 1:length(t)

# Propagate the orbit.
r_teme, v_teme = sgp4!(sgp4d, t[k])

# Assemble the result vector.
st_sgp4_result = vcat(t[k], r_teme, v_teme)

# Compare the values.
@test t[k] == SGP4_results[k, 1]
@test r_teme[1] ≈ SGP4_results[k, 2] atol=1e-8
@test r_teme[2] ≈ SGP4_results[k, 3] atol=1e-8
@test r_teme[3] ≈ SGP4_results[k, 4] atol=1e-8
@test v_teme[1] ≈ SGP4_results[k, 5] atol=1e-9
@test v_teme[2] ≈ SGP4_results[k, 6] atol=1e-9
@test v_teme[3] ≈ SGP4_results[k, 7] atol=1e-9
end
end
end

@testset "Simultaneous Creation and Propagation" begin
# Read all TLEs that will be used to test.
tles = read_tles_from_file("./sgp4_tests/sgp4_tests.tle")

for tle in tles
filename = @sprintf(
"./sgp4_tests/aiaa-2006-6753/sgp4_tle_%d_result.txt",
tle.satellite_number
)
SGP4_results = readdlm(filename; comments = true)
t = SGP4_results[:,1]

# Initialize the orbit propagator.
r_teme, v_teme, sgp4d = sgp4(t[end], tle; sgp4c = sgp4c_wgs72)

# We test just the final instant to save computational burden.
@test t[end] == SGP4_results[end, 1]
@test r_teme[1] ≈ SGP4_results[end, 2] atol=1e-8
@test r_teme[2] ≈ SGP4_results[end, 3] atol=1e-8
@test r_teme[3] ≈ SGP4_results[end, 4] atol=1e-8
@test v_teme[1] ≈ SGP4_results[end, 5] atol=1e-9
@test v_teme[2] ≈ SGP4_results[end, 6] atol=1e-9
@test v_teme[3] ≈ SGP4_results[end, 7] atol=1e-9
end
end
end

@testset "Errors" begin
tle = tle"""
AMAZONIA 1
1 47699U 21015A 23083.68657856 -.00000044 10000-8 43000-4 0 9990
2 47699 98.4304 162.1097 0001247 136.2017 223.9283 14.40814394108652
"""

sgp4d = sgp4_init(tle)
sgp4d.algorithm = :any
@test_throws ErrorException sgp4!(sgp4d, 10)
@testset "SGP4 TLEs" verbose = true begin
include("./tle.jl")
end
149 changes: 149 additions & 0 deletions test/sgp4.jl
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,149 @@
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
#
# Description
# ==========================================================================================
#
# Test SGP4 algorithm. All tests are based on [1].
#
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
#
# References
# ==========================================================================================
#
# [1] Vallado, D. A., Crawford, P., Hujsak, R., Kelso, T. S (2006). Revisiting Spacetrack
# Report #3: Rev1. AIAA.
#
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #

@testset "Constructors" begin
sgp4ds = SatelliteToolboxSgp4.Sgp4DeepSpace{Float64}(
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, false, false, false,
)

sgp4c = Sgp4Propagator{Float64, Float64}(
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, :sdp4, sgp4c_wgs84, sgp4ds
)

# Some random tests.
@test sgp4c.epoch == 0.0
@test sgp4c.a_k == 0.0
@test sgp4c.∂M == 0.0
@test sgp4c.algorithm == :sdp4
@test sgp4c.sgp4ds == sgp4ds
end

@testset "Tests from the Paper AIAA 2006-6753" verbose = true begin
# Read all TLEs that will be used to test.
tles = read_tles_from_file("./sgp4_tests/sgp4_tests.tle")

@testset "Default" begin
for tle in tles
filename = @sprintf(
"./sgp4_tests/aiaa-2006-6753/sgp4_tle_%d_result.txt",
tle.satellite_number
)
SGP4_results = readdlm(filename; comments = true)

# Initialize the orbit propagator.
sgp4d = sgp4_init(tle; sgp4c = sgp4c_wgs72)

t = SGP4_results[:, 1]

@inbounds for k in 1:length(t)

# Propagate the orbit.
r_teme, v_teme = sgp4!(sgp4d, t[k])

# Assemble the result vector.
st_sgp4_result = vcat(t[k], r_teme, v_teme)

# Compare the values.
@test t[k] == SGP4_results[k, 1]
@test r_teme[1] ≈ SGP4_results[k, 2] atol=1e-8
@test r_teme[2] ≈ SGP4_results[k, 3] atol=1e-8
@test r_teme[3] ≈ SGP4_results[k, 4] atol=1e-8
@test v_teme[1] ≈ SGP4_results[k, 5] atol=1e-9
@test v_teme[2] ≈ SGP4_results[k, 6] atol=1e-9
@test v_teme[3] ≈ SGP4_results[k, 7] atol=1e-9
end
end
end

@testset "In-place Initialization" begin
# First, we create a dummy SGP4 structure but with the correct constants and epoch
# type.
sgp4d = sgp4_init(0.0, 0, 0, 0, 0, 0, 0, 0; sgp4c = sgp4c_wgs72)

for tle in tles
filename = @sprintf(
"./sgp4_tests/aiaa-2006-6753/sgp4_tle_%d_result.txt",
tle.satellite_number
)
SGP4_results = readdlm(filename; comments = true)

# Initialize the orbit propagator.
sgp4_init!(sgp4d, tle)

t = SGP4_results[:,1]

@inbounds for k in 1:length(t)

# Propagate the orbit.
r_teme, v_teme = sgp4!(sgp4d, t[k])

# Assemble the result vector.
st_sgp4_result = vcat(t[k], r_teme, v_teme)

# Compare the values.
@test t[k] == SGP4_results[k, 1]
@test r_teme[1] ≈ SGP4_results[k, 2] atol=1e-8
@test r_teme[2] ≈ SGP4_results[k, 3] atol=1e-8
@test r_teme[3] ≈ SGP4_results[k, 4] atol=1e-8
@test v_teme[1] ≈ SGP4_results[k, 5] atol=1e-9
@test v_teme[2] ≈ SGP4_results[k, 6] atol=1e-9
@test v_teme[3] ≈ SGP4_results[k, 7] atol=1e-9
end
end
end

@testset "Simultaneous Creation and Propagation" begin
# Read all TLEs that will be used to test.
tles = read_tles_from_file("./sgp4_tests/sgp4_tests.tle")

for tle in tles
filename = @sprintf(
"./sgp4_tests/aiaa-2006-6753/sgp4_tle_%d_result.txt",
tle.satellite_number
)
SGP4_results = readdlm(filename; comments = true)
t = SGP4_results[:,1]

# Initialize the orbit propagator.
r_teme, v_teme, sgp4d = sgp4(t[end], tle; sgp4c = sgp4c_wgs72)

# We test just the final instant to save computational burden.
@test t[end] == SGP4_results[end, 1]
@test r_teme[1] ≈ SGP4_results[end, 2] atol=1e-8
@test r_teme[2] ≈ SGP4_results[end, 3] atol=1e-8
@test r_teme[3] ≈ SGP4_results[end, 4] atol=1e-8
@test v_teme[1] ≈ SGP4_results[end, 5] atol=1e-9
@test v_teme[2] ≈ SGP4_results[end, 6] atol=1e-9
@test v_teme[3] ≈ SGP4_results[end, 7] atol=1e-9
end
end
end

@testset "Errors" begin
tle = tle"""
AMAZONIA 1
1 47699U 21015A 23083.68657856 -.00000044 10000-8 43000-4 0 9990
2 47699 98.4304 162.1097 0001247 136.2017 223.9283 14.40814394108652
"""

sgp4d = sgp4_init(tle)
sgp4d.algorithm = :any
@test_throws ErrorException sgp4!(sgp4d, 10)
end

0 comments on commit fc735b5

Please sign in to comment.