def test_compare_zonal_to_coe(self):
X0_mee = mee_0
X0_coe = coe_0
order_H = 6
kep_dyn_mee = mee.KeplerianDynamics()
zon_grav_mee = mee.ZonalGravity(order=order_H)
sysmee = utl.SystemDynamics(kep_dyn_mee, zon_grav_mee)
kep_dyn_coe = coe.KeplerianDynamics()
zon_grav_coe = coe.ZonalGravity(order=order_H)
syscoe = utl.SystemDynamics(kep_dyn_coe, zon_grav_coe)
segs_per_orbit = 6
orbits = 1
segments = orbits * segs_per_orbit
domains = [k * period / segs_per_orbit for k in range(segments + 1)]
seg_number = len(domains) - 1
N = (order_mcpi, ) * seg_number
mcpi_mee = mcpyi.MCPI(sysmee, domains, N, 'warm', X0_mee, tol)
mcpi_coe = mcpyi.MCPI(syscoe, domains, N, 'warm', X0_coe, tol)
X_mee = mcpi_mee.solve_serial()(T)
X_coe = mcpi_coe.solve_serial()(T)
diff = convert.mod_angles(np.abs(X_coe - convert.coe_mee(X_mee)))
indices_2pi = np.where(2 * np.pi - tol < diff)
diff[indices_2pi] -= 2 * np.pi
np.testing.assert_allclose(diff, 0., rtol=0, atol=tol)
def test_compare_constant_thrust_to_rv(self):
X0_coe = coe_0
X0_rv = rv_0
u = np.array([[0., 1e-6, 0.]])
kep_dyn_coe = coe.KeplerianDynamics()
conthrust_coe = coe.ConstantThrust(u)
syscoe = utl.SystemDynamics(kep_dyn_coe, conthrust_coe)
kep_dyn_rv = rv.KeplerianDynamics()
conthrust_rv = rv.ConstantThrust(u)
sysrv = utl.SystemDynamics(kep_dyn_rv, conthrust_rv)
segments = orbits * segs_per_orbit
domains = [k * period / segs_per_orbit for k in range(segments + 1)]
seg_number = len(domains) - 1
N = (order_mcpi, ) * seg_number
mcpi_coe = mcpyi.MCPI(syscoe, domains, N, 'warm', X0_coe, tol)
mcpi_rv = mcpyi.MCPI(sysrv, domains, N, 'warm', X0_rv, tol)
X_coe = mcpi_coe.solve_serial()(T)
X_rv = mcpi_rv.solve_serial()(T)
np.testing.assert_allclose(X_rv,
convert.rv_coe(X_coe),
rtol=0,
atol=tol * 10)
def test_compare_a_eci_to_rv(self):
X0_coe = coe_0
X0_rv = rv_0
order_H = 6
kep_dyn_coe = coe.KeplerianDynamics()
zon_grav_coe = coe.ZonalGravity(order=order_H)
syscoe = utl.SystemDynamics(kep_dyn_coe, zon_grav_coe)
kep_dyn_rv = rv.KeplerianDynamics()
zon_grav_rv = rv.ZonalGravity(order=order_H)
sysrv = utl.SystemDynamics(kep_dyn_rv, zon_grav_rv)
segs_per_orbit = 6
segments = orbits * segs_per_orbit
domains = [k * period / segs_per_orbit for k in range(segments + 1)]
seg_number = len(domains) - 1
N = (order_mcpi, ) * seg_number
mcpi_coe = mcpyi.MCPI(syscoe, domains, N, 'warm', X0_coe, tol)
mcpi_rv = mcpyi.MCPI(sysrv, domains, N, 'warm', X0_rv, tol)
mcpi_coe.solve_serial()(T)
mcpi_rv.solve_serial()(T)
np.testing.assert_allclose(zon_grav_coe.a_eci,
zon_grav_rv.a_eci,
rtol=0,
atol=tol)
def test_hamiltonian_dynamics(self):
X0 = coe_0
kep_dyn = coe.KeplerianDynamics()
segments = orbits * segs_per_orbit
domains = [k * period / segs_per_orbit for k in range(segments + 1)]
seg_number = len(domains) - 1
N = (order_mcpi, ) * seg_number
mcpi = mcpyi.MCPI(kep_dyn, domains, N, 'warm', X0, tol)
X = mcpi.solve_serial()(T)
order_H = 1
H = coe.Hamiltonian(order=order_H)(T, X)
np.testing.assert_allclose(H[0, 0], H, rtol=tol)
def test_zonal_gravity(self):
X0 = coe_0
order_H = 6
kep_dyn = coe.KeplerianDynamics()
zon_grav = coe.ZonalGravity(order=order_H)
system = utl.SystemDynamics(kep_dyn, perturbations=zon_grav)
segments = orbits * segs_per_orbit
domains = [k * period / segs_per_orbit for k in range(segments + 1)]
seg_number = len(domains) - 1
N = (order_mcpi, ) * seg_number
mcpi = mcpyi.MCPI(system, domains, N, 'warm', X0, tol)
X = mcpi.solve_serial()(T)
H = coe.Hamiltonian(order=order_H)(T, X)
np.testing.assert_allclose(H[0, 0], H, rtol=tol)
def test_compare_dynamics_to_rv(self):
X0_coe = coe_0
X0_rv = rv_0
coe_dyn = coe.KeplerianDynamics()
rv_dyn = rv.KeplerianDynamics()
segments = orbits * segs_per_orbit
domains = [k * period / segs_per_orbit for k in range(segments + 1)]
seg_number = len(domains) - 1
N = (order_mcpi, ) * seg_number
mcpi_coe = mcpyi.MCPI(coe_dyn, domains, N, 'warm', X0_coe, tol)
mcpi_rv = mcpyi.MCPI(rv_dyn, domains, N, 'warm', X0_rv, tol)
X_coe = mcpi_coe.solve_serial()(T)
X_rv = mcpi_rv.solve_serial()(T)
np.testing.assert_allclose(X_rv,
convert.rv_coe(X_coe),
rtol=0,
atol=1e-13)