open-space-toolkit-astrodynamics 2.2.0__py311-none-any.whl → 5.0.0__py311-none-any.whl

ostk/astrodynamics/test/trajectory/test_propagator.py

@@ -1,234 +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.time import Duration
-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.trajectory.state 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.event_condition import InstantCondition
-
-
-@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(state: State) -> InstantCondition:
-    return InstantCondition(
-        InstantCondition.Criterion.StrictlyPositive,
-        state.get_instant() + Duration.seconds(42.0),
-    )
-
-
-@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_access_numerical_solver(
-        self, propagator: Propagator, numerical_solver: NumericalSolver
-    ):
-        assert propagator.access_numerical_solver() == numerical_solver
-
-    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: InstantCondition,
-    ):
-        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,121 +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
-from ostk.astrodynamics.trajectory.state import CoordinatesBroker
-from ostk.astrodynamics.trajectory.state.coordinates_subset import (
-    CartesianPosition,
-    CartesianVelocity,
-)
-
-
-@pytest.fixture()
-def instant() -> Instant:
-    return Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-
-
-@pytest.fixture
-def frame() -> Frame:
-    return Frame.GCRF()
-
-
-@pytest.fixture()
-def position(frame: Frame) -> Position:
-    return Position.meters([6371000.0, 0.0, 0.0], frame)
-
-
-@pytest.fixture()
-def velocity(frame: Frame) -> Velocity:
-    return Velocity.meters_per_second([7600.0, 0.0, 0.0], frame)
-
-
-@pytest.fixture
-def state(
-    instant: Instant, position: Position, velocity: Velocity, frame: Frame
-) -> State:
-    return State(instant, position, velocity)
-
-
-@pytest.fixture
-def coordinates_broker() -> CoordinatesBroker:
-    return CoordinatesBroker([CartesianPosition.default(), CartesianVelocity.default()])
-
-
-class TestState:
-    def test_constructor(
-        self,
-        instant: Instant,
-        position: Position,
-        velocity: Velocity,
-    ):
-        state = State(instant, position, velocity)
-        assert state is not None
-        assert isinstance(state, State)
-        assert state.is_defined()
-
-    def test_explicit_constructor(
-        self,
-        instant: Instant,
-        position: Position,
-        velocity: Velocity,
-        frame: Frame,
-        coordinates_broker: CoordinatesBroker,
-    ):
-        state = State(instant, position, velocity)
-        assert state is not None
-        assert isinstance(state, State)
-        assert state.is_defined()
-
-        state = State(
-            instant,
-            np.append(position.get_coordinates(), velocity.get_coordinates()),
-            frame,
-            coordinates_broker,
-        )
-
-        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,
-        instant: Instant,
-        position: Position,
-        velocity: Velocity,
-        frame: Frame,
-    ):
-        assert state.get_instant() == instant
-        assert state.get_position() == position
-        assert state.get_velocity() == velocity
-        assert state.get_frame() == frame
-        assert (
-            state.get_coordinates()
-            == np.append(position.get_coordinates(), velocity.get_coordinates())
-        ).all()
-
-    def test_in_frame(
-        self,
-        state: State,
-        frame: Frame,
-    ):
-        assert state.in_frame(frame) == state
-        assert state.in_frame(Frame.ITRF()) != state