open-space-toolkit-astrodynamics 1.2.0__py310-none-any.whl → 4.2.0__py310-none-any.whl

ostk/astrodynamics/test/trajectory/test_orbit.py

@@ -1,92 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.physics import Environment
-from ostk.physics.units import Length, Angle
-from ostk.physics.time import Scale, Instant, DateTime, Time
-
-from ostk.astrodynamics.trajectory import Orbit, State
-from ostk.astrodynamics.trajectory.orbit.models import SGP4
-from ostk.astrodynamics.trajectory.orbit.models.sgp4 import TLE
-
-
-@pytest.fixture
-def earth():
-    return Environment.default().access_celestial_object_with_name("Earth")
-
-
-class TestOrbit:
-    def test_trajectory_orbit_constructors(self, earth):
-        # Construct Two-Line Element set
-        tle = TLE(
-            "1 25544U 98067A   18231.17878740  .00000187  00000-0  10196-4 0  9994",
-            "2 25544  51.6447  64.7824 0005971  73.1467  36.4366 15.53848234128316",
-        )
-
-        # Construct orbit using SGP4 model
-        orbit = Orbit(SGP4(tle), earth)
-
-        assert orbit is not None
-        assert isinstance(orbit, Orbit)
-        assert orbit.is_defined()
-
-        # Construct get state at current epoch
-        state: State = orbit.get_state_at(Instant.now())
-
-        assert state is not None
-        assert isinstance(state, State)
-
-    def test_trajectory_orbit_circular(self, earth):
-        epoch = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-        altitude = Length.kilometers(500.0)
-        inclination = Angle.degrees(45.0)
-
-        orbit: Orbit = Orbit.circular(epoch, altitude, inclination, earth)
-
-    def test_trajectory_orbit_equatorial(self, earth):
-        epoch = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-        apoapsis_altitude = Length.kilometers(500.1)
-        periapsis_altitude = Length.kilometers(499.9)
-
-        orbit: Orbit = Orbit.equatorial(
-            epoch, apoapsis_altitude, periapsis_altitude, earth
-        )
-
-        assert orbit is not None
-        assert isinstance(orbit, Orbit)
-        assert orbit.is_defined()
-
-    def test_trajectory_orbit_circular_equatorial(self, earth):
-        epoch = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-        altitude = Length.kilometers(500.0)
-
-        orbit: Orbit = Orbit.circular_equatorial(epoch, altitude, earth)
-
-        assert orbit is not None
-        assert isinstance(orbit, Orbit)
-        assert orbit.is_defined()
-
-    def test_trajectory_orbit_geo_synchronous(self, earth):
-        epoch = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-        inclination = Angle.degrees(45.0)
-        longitude = Angle.degrees(45.0)
-
-        orbit: Orbit = Orbit.geo_synchronous(epoch, inclination, longitude, earth)
-
-        assert orbit is not None
-        assert isinstance(orbit, Orbit)
-        assert orbit.is_defined()
-
-    def test_trajectory_orbit_sun_synchronous(self, earth):
-        epoch = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-        altitude = Length.kilometers(500.0)
-        local_time_at_descending_node = Time.midnight()
-
-        orbit: Orbit = Orbit.sun_synchronous(
-            epoch, altitude, local_time_at_descending_node, earth
-        )
-
-        assert orbit is not None
-        assert isinstance(orbit, Orbit)
-        assert orbit.is_defined()

ostk/astrodynamics/test/trajectory/test_propagator.py

@@ -1,233 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-import numpy as np
-
-from ostk.mathematics.geometry.d3.objects import Cuboid
-from ostk.mathematics.geometry.d3.objects import Composite
-from ostk.mathematics.geometry.d3.objects import Point
-
-from ostk.physics import Environment
-from ostk.physics.units import Mass
-from ostk.physics.time import Instant
-from ostk.physics.time import DateTime
-from ostk.physics.time import Scale
-from ostk.physics.coordinate import Position
-from ostk.physics.coordinate import Velocity
-from ostk.physics.coordinate import Frame
-from ostk.physics.environment.objects.celestial_bodies import Earth
-
-from ostk.astrodynamics import NumericalSolver
-from ostk.astrodynamics.flight.system import SatelliteSystem
-from ostk.astrodynamics.flight.system import Dynamics
-from ostk.astrodynamics.flight.system.dynamics import CentralBodyGravity
-from ostk.astrodynamics.flight.system.dynamics import PositionDerivative
-from ostk.astrodynamics.trajectory import State
-from ostk.astrodynamics.trajectory import Propagator
-from ostk.astrodynamics import EventCondition
-
-
-@pytest.fixture
-def satellite_system() -> SatelliteSystem:
-    mass = Mass(90.0, Mass.Unit.Kilogram)
-    satellite_geometry = Composite(
-        Cuboid(
-            Point(0.0, 0.0, 0.0),
-            [[1.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 0.0, 1.0]],
-            [1.0, 0.0, 0.0],
-        )
-    )
-    inertia_tensor = np.identity(3)
-    surface_area = 0.8
-    drag_coefficient = 0.0
-
-    return SatelliteSystem(
-        mass, satellite_geometry, inertia_tensor, surface_area, drag_coefficient
-    )
-
-
-@pytest.fixture
-def environment() -> Environment:
-    return Environment.default()
-
-
-@pytest.fixture
-def state() -> State:
-    frame: Frame = Frame.GCRF()
-    position: Position = Position.meters([7500000.0, 0.0, 0.0], frame)
-    velocity: Velocity = Velocity.meters_per_second(
-        [0.0, 5335.865450622126, 5335.865450622126], frame
-    )
-
-    instant: Instant = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-    return State(instant, position, velocity)
-
-
-@pytest.fixture
-def central_body_gravity() -> CentralBodyGravity:
-    return CentralBodyGravity(Earth.WGS84(20, 0))
-
-
-@pytest.fixture
-def position_derivative() -> PositionDerivative:
-    return PositionDerivative()
-
-
-@pytest.fixture
-def dynamics(
-    position_derivative: PositionDerivative, central_body_gravity: CentralBodyGravity
-) -> list:
-    return [position_derivative, central_body_gravity]
-
-
-@pytest.fixture
-def numerical_solver() -> NumericalSolver:
-    return NumericalSolver(
-        NumericalSolver.LogType.NoLog,
-        NumericalSolver.StepperType.RungeKuttaFehlberg78,
-        5.0,
-        1.0e-15,
-        1.0e-15,
-    )
-
-
-@pytest.fixture
-def conditional_numerical_solver() -> NumericalSolver:
-    return NumericalSolver(
-        NumericalSolver.LogType.NoLog,
-        NumericalSolver.StepperType.RungeKuttaDopri5,
-        5.0,
-        1.0e-15,
-        1.0e-15,
-    )
-
-
-@pytest.fixture
-def event_condition() -> EventCondition:
-    class MyEventCondition(EventCondition):
-        def evaluate(self, state_vector, time):
-            return time - 42.0
-
-    return MyEventCondition("42 Seconds", EventCondition.Criteria.StrictlyPositive)
-
-
-@pytest.fixture
-def propagator(
-    numerical_solver: NumericalSolver, dynamics: list[Dynamics]
-) -> Propagator:
-    return Propagator(numerical_solver, dynamics)
-
-
-class TestPropagator:
-    def test_constructors(self, propagator: Propagator):
-        assert propagator is not None
-        assert isinstance(propagator, Propagator)
-        assert propagator.is_defined()
-
-    def test_get_dynamics(self, propagator: Propagator, dynamics: list):
-        assert propagator.get_dynamics() == dynamics
-
-    def test_set_dynamics(self, propagator: Propagator, dynamics: list):
-        assert len(propagator.get_dynamics()) == 2
-
-        propagator.set_dynamics(dynamics + dynamics)
-
-        assert len(propagator.get_dynamics()) == 4
-
-    def test_add_dynamics(
-        self, propagator: Propagator, central_body_gravity: CentralBodyGravity
-    ):
-        assert len(propagator.get_dynamics()) == 2
-
-        propagator.add_dynamics(central_body_gravity)
-        propagator.add_dynamics(central_body_gravity)
-
-        assert len(propagator.get_dynamics()) == 4
-
-    def test_clear_dynamics(self, propagator: Propagator):
-        assert len(propagator.get_dynamics()) >= 1
-
-        propagator.clear_dynamics()
-
-        assert len(propagator.get_dynamics()) == 0
-
-    def test_calculate_state_at(self, propagator: Propagator, state: State):
-        instant: Instant = Instant.date_time(DateTime(2018, 1, 1, 0, 10, 0), Scale.UTC)
-
-        propagator_state = propagator.calculate_state_at(state, instant)
-
-        propagator_state_position_ref = np.array(
-            [6265892.25765909, 3024770.94961259, 3024359.72137468]
-        )
-        propagator_state_velocity_ref = np.array(
-            [-3974.49168221, 4468.16996776, 4466.19232746]
-        )
-
-        propagator_state_position = propagator_state.get_position().get_coordinates()
-        propagator_state_velocity = propagator_state.get_velocity().get_coordinates()
-
-        assert all(
-            [
-                round(propagator_state_position[i], 8)
-                == round(propagator_state_position_ref[i], 8)
-                for i in range(0, len(propagator_state_position_ref))
-            ]
-        )
-        assert all(
-            [
-                round(propagator_state_velocity[i], 8)
-                == round(propagator_state_velocity_ref[i], 8)
-                for i in range(0, len(propagator_state_velocity_ref))
-            ]
-        )
-        assert propagator_state.get_instant() == instant
-
-    def test_calculate_state_at(
-        self,
-        conditional_numerical_solver: NumericalSolver,
-        dynamics: list[Dynamics],
-        state: State,
-        event_condition: EventCondition,
-    ):
-        propagator: Propagator = Propagator(conditional_numerical_solver, dynamics)
-
-        instant: Instant = Instant.date_time(DateTime(2018, 1, 1, 0, 10, 0), Scale.UTC)
-
-        propagator_state = propagator.calculate_state_at(
-            state, instant, event_condition
-        )
-
-        assert pytest.approx(42.0, abs=1e-3) == float(
-            (propagator_state.get_instant() - state.get_instant()).in_seconds()
-        )
-
-    def test_calculate_states_at(self, propagator: Propagator, state: State):
-        instant_array = [
-            Instant.date_time(DateTime(2018, 1, 1, 0, 10, 0), Scale.UTC),
-            Instant.date_time(DateTime(2018, 1, 1, 0, 20, 0), Scale.UTC),
-        ]
-
-        _ = propagator.calculate_states_at(state, instant_array)
-
-        with pytest.raises(RuntimeError):
-            instant_array.reverse()
-            propagator.calculate_states_at(state, instant_array)
-
-    def test_from_environment(
-        self,
-        numerical_solver: NumericalSolver,
-        environment: Environment,
-        satellite_system: SatelliteSystem,
-    ):
-        assert Propagator.from_environment(numerical_solver, environment) is not None
-
-        assert (
-            Propagator.from_environment(numerical_solver, environment, satellite_system)
-            is not None
-        )
-
-    def test_default(self, environment: Environment, satellite_system: SatelliteSystem):
-        assert Propagator.default()
-        assert Propagator.default(environment) is not None
-        assert Propagator.default(environment, satellite_system) is not None

ostk/astrodynamics/test/trajectory/test_state.py

@@ -1,70 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-import numpy as np
-
-from ostk.physics.time import Instant
-from ostk.physics.time import DateTime
-from ostk.physics.time import Scale
-from ostk.physics.coordinate import Position
-from ostk.physics.coordinate import Velocity
-from ostk.physics.coordinate import Frame
-
-from ostk.astrodynamics.trajectory import State
-
-
-@pytest.fixture
-def state_default_inputs() -> tuple[Instant, Position, Velocity, Frame]:
-    frame: Frame = Frame.GCRF()
-    position: Position = Position.meters([6371000.0, 0.0, 0.0], frame)
-    velocity: Velocity = Velocity.meters_per_second([7600.0, 0.0, 0.0], frame)
-
-    instant = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-
-    return (instant, position, velocity, frame)
-
-
-@pytest.fixture
-def state(state_default_inputs) -> State:
-    return State(*state_default_inputs[0:-1])
-
-
-class TestState:
-    def test_constructor(self, state: State):
-        assert state is not None
-        assert isinstance(state, State)
-        assert state.is_defined()
-
-    def test_comparators(self, state: State):
-        assert (state == state) is True
-        assert (state != state) is False
-
-    def test_operators(self, state: State):
-        assert isinstance(state + state, State)
-        assert isinstance(state - state, State)
-
-    def test_getters(
-        self,
-        state: State,
-        state_default_inputs: tuple[Instant, Position, Velocity, Frame],
-    ):
-        (instant, position, velocity, _) = state_default_inputs
-
-        assert state.get_instant() == instant
-        assert state.get_position() == position
-        assert state.get_velocity() == velocity
-        assert (
-            state.get_coordinates()
-            == np.append(position.get_coordinates(), velocity.get_coordinates())
-        ).all()
-
-    def test_in_frame(
-        self,
-        state: State,
-        state_default_inputs: tuple[Instant, Position, Velocity, Frame],
-    ):
-        (_, _, _, frame) = state_default_inputs
-
-        assert state.in_frame(frame) == state
-        assert state.in_frame(Frame.ITRF()) != state