Browse Source

refactor: test_true_anomaly_roundtrip - SCENARIO/SECTION pattern, tight tolerances

test-refactor
cinnaboot 3 months ago
parent
commit
db7094abed
  1. 130
      tests/test_true_anomaly_roundtrip.cpp

130
tests/test_true_anomaly_roundtrip.cpp

@ -4,140 +4,56 @@
#include "../src/orbital_mechanics.h" #include "../src/orbital_mechanics.h"
#include <cmath> #include <cmath>
SCENARIO("True anomaly round-trip conversion preserves values", using Catch::Matchers::WithinAbs;
"[orbital_elements][true_anomaly]") {
SCENARIO("True anomaly round-trip conversion and radius sanity checks",
"[orbital_elements][true_anomaly][sanity]") {
const double parent_mass = 5.972e24; const double parent_mass = 5.972e24;
const double a = 7000e3; const double a = 7000e3;
const double e = 0.3; const double e = 0.3;
OrbitalElements elements = {0};
elements.semi_major_axis = a;
elements.eccentricity = e;
Vec3 pos, vel;
SECTION("at periapsis (nu = 0)") { // Precomputed analytical values
const double expected_nu = 0.0; const double expected_r_peri = a * (1.0 - e); // 4900000.0
const double expected_r_apo = a * (1.0 + e); // 9100000.0
OrbitalElements elements = {0};
elements.semi_major_axis = a;
elements.eccentricity = e;
elements.true_anomaly = expected_nu;
elements.inclination = 0.0;
elements.longitude_of_ascending_node = 0.0;
elements.argument_of_periapsis = 0.0;
Vec3 pos, vel;
orbital_elements_to_cartesian(elements, parent_mass, &pos, &vel);
OrbitalElements reconstructed = cartesian_to_orbital_elements(pos, vel, parent_mass);
INFO("Expected nu: " << expected_nu);
INFO("Reconstructed: " << reconstructed.true_anomaly);
REQUIRE_THAT(reconstructed.true_anomaly,
Catch::Matchers::WithinAbs(expected_nu, 1e-12));
}
SECTION("at apoapsis (nu = pi)") {
const double expected_nu = M_PI;
OrbitalElements elements = {0};
elements.semi_major_axis = a;
elements.eccentricity = e;
elements.true_anomaly = expected_nu;
elements.inclination = 0.0;
elements.longitude_of_ascending_node = 0.0;
elements.argument_of_periapsis = 0.0;
Vec3 pos, vel;
orbital_elements_to_cartesian(elements, parent_mass, &pos, &vel);
OrbitalElements reconstructed = cartesian_to_orbital_elements(pos, vel, parent_mass);
INFO("Expected nu: " << expected_nu);
INFO("Reconstructed: " << reconstructed.true_anomaly);
REQUIRE_THAT(reconstructed.true_anomaly,
Catch::Matchers::WithinAbs(expected_nu, 1e-12));
}
SECTION("at 90 degrees (nu = pi/2)") {
const double expected_nu = M_PI / 2.0;
OrbitalElements elements = {0};
elements.semi_major_axis = a;
elements.eccentricity = e;
elements.true_anomaly = expected_nu;
elements.inclination = 0.0;
elements.longitude_of_ascending_node = 0.0;
elements.argument_of_periapsis = 0.0;
Vec3 pos, vel;
orbital_elements_to_cartesian(elements, parent_mass, &pos, &vel);
OrbitalElements reconstructed = cartesian_to_orbital_elements(pos, vel, parent_mass);
INFO("Expected nu: " << expected_nu);
INFO("Reconstructed: " << reconstructed.true_anomaly);
REQUIRE_THAT(reconstructed.true_anomaly,
Catch::Matchers::WithinAbs(expected_nu, 1e-12));
}
SECTION("at 270 degrees (nu = 3*pi/2)") {
const double expected_nu = 3.0 * M_PI / 2.0;
OrbitalElements elements = {0}; auto check_roundtrip = [&](double expected_nu) {
elements.semi_major_axis = a;
elements.eccentricity = e;
elements.true_anomaly = expected_nu; elements.true_anomaly = expected_nu;
elements.inclination = 0.0;
elements.longitude_of_ascending_node = 0.0;
elements.argument_of_periapsis = 0.0;
Vec3 pos, vel;
orbital_elements_to_cartesian(elements, parent_mass, &pos, &vel); orbital_elements_to_cartesian(elements, parent_mass, &pos, &vel);
OrbitalElements reconstructed = cartesian_to_orbital_elements(pos, vel, parent_mass); OrbitalElements reconstructed = cartesian_to_orbital_elements(pos, vel, parent_mass);
INFO("Expected nu: " << expected_nu); INFO("Expected nu: " << expected_nu);
INFO("Reconstructed: " << reconstructed.true_anomaly); INFO("Reconstructed: " << reconstructed.true_anomaly);
REQUIRE_THAT(reconstructed.true_anomaly, WithinAbs(expected_nu, 1e-12));
};
REQUIRE_THAT(reconstructed.true_anomaly, SECTION("at periapsis (nu = 0)") { check_roundtrip(0.0); }
Catch::Matchers::WithinAbs(expected_nu, 1e-12)); SECTION("at apoapsis (nu = pi)") { check_roundtrip(M_PI); }
} SECTION("at 90 degrees (nu = pi/2)") { check_roundtrip(M_PI / 2.0); }
} SECTION("at 270 degrees (nu = 3*pi/2)") { check_roundtrip(3.0 * M_PI / 2.0); }
SCENARIO("Radius at periapsis and apoapsis matches analytical formula",
"[orbital_elements][sanity]") {
const double parent_mass = 5.972e24;
const double a = 7000e3;
const double e = 0.3;
const double expected_r_peri = a * (1.0 - e); // 4900000.0
const double expected_r_apo = a * (1.0 + e); // 9100000.0
SECTION("periapsis radius = a*(1-e)") { SECTION("periapsis radius = a*(1-e)") {
OrbitalElements peri = {0}; orbital_elements_to_cartesian(elements, parent_mass, &pos, &vel);
peri.semi_major_axis = a;
peri.eccentricity = e;
peri.true_anomaly = 0.0;
Vec3 pos, vel;
orbital_elements_to_cartesian(peri, parent_mass, &pos, &vel);
const double r_peri = vec3_magnitude(pos); const double r_peri = vec3_magnitude(pos);
INFO("Expected r: " << expected_r_peri); INFO("Expected r: " << expected_r_peri);
INFO("Calculated r: " << r_peri); INFO("Calculated r: " << r_peri);
REQUIRE_THAT(r_peri, REQUIRE_THAT(r_peri, WithinAbs(expected_r_peri, 1e-6));
Catch::Matchers::WithinAbs(expected_r_peri, 1e-6));
} }
SECTION("apoapsis radius = a*(1+e)") { SECTION("apoapsis radius = a*(1+e)") {
OrbitalElements apo = {0}; elements.true_anomaly = M_PI;
apo.semi_major_axis = a;
apo.eccentricity = e;
apo.true_anomaly = M_PI;
Vec3 pos, vel; orbital_elements_to_cartesian(elements, parent_mass, &pos, &vel);
orbital_elements_to_cartesian(apo, parent_mass, &pos, &vel);
const double r_apo = vec3_magnitude(pos); const double r_apo = vec3_magnitude(pos);
INFO("Expected r: " << expected_r_apo); INFO("Expected r: " << expected_r_apo);
INFO("Calculated r: " << r_apo); INFO("Calculated r: " << r_apo);
REQUIRE_THAT(r_apo, WithinAbs(expected_r_apo, 1e-6));
REQUIRE_THAT(r_apo,
Catch::Matchers::WithinAbs(expected_r_apo, 1e-6));
} }
} }

Loading…
Cancel
Save