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

ostk/astrodynamics/test/event_condition/test_logical_condition.py

@@ -1,120 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.physics.coordinate import Frame, Position, Velocity
-from ostk.physics.time import DateTime, Instant, Scale
-
-from ostk.astrodynamics.event_condition import RealCondition, LogicalCondition
-from ostk.astrodynamics.trajectory import State
-
-
-@pytest.fixture
-def always_true() -> RealCondition:
-    return RealCondition(
-        "Always True Condition",
-        RealCondition.Criterion.StrictlyPositive,
-        lambda state: 1.0,
-        0.0,
-    )
-
-
-@pytest.fixture
-def always_false() -> RealCondition:
-    return RealCondition(
-        "Always False Condition",
-        RealCondition.Criterion.StrictlyPositive,
-        lambda state: -1.0,
-        0.0,
-    )
-
-
-@pytest.fixture
-def event_conditions(
-    always_true: RealCondition, always_false: RealCondition
-) -> list[RealCondition]:
-    return [always_true, always_false]
-
-
-@pytest.fixture
-def logical_condition_type() -> LogicalCondition.Type:
-    return LogicalCondition.Type.And
-
-
-@pytest.fixture
-def state() -> State:
-    frame: Frame = Frame.GCRF()
-    position: Position = Position.meters(
-        [717094.039086306, -6872433.2241124, 46175.9696673281], frame
-    )
-    velocity: Velocity = Velocity.meters_per_second(
-        [-970.650826004612, -45.4598114773158, 7529.82424886455], frame
-    )
-
-    instant: Instant = Instant.date_time(DateTime(2023, 1, 1, 0, 1, 0), Scale.UTC)
-    return State(instant, position, velocity)
-
-
-@pytest.fixture
-def logical_condition(
-    logical_condition_type: LogicalCondition.Type, event_conditions: list[RealCondition]
-) -> LogicalCondition:
-    return LogicalCondition("Logical", logical_condition_type, event_conditions)
-
-
-class TestLogicalCondition:
-    @pytest.mark.parametrize(
-        "logical_condition_type",
-        (
-            LogicalCondition.Type.And,
-            LogicalCondition.Type.Or,
-        ),
-    )
-    def test_constructor(
-        self,
-        logical_condition_type: LogicalCondition.Type,
-        event_conditions: list[RealCondition],
-    ):
-        assert (
-            LogicalCondition(
-                "Logical Condition", logical_condition_type, event_conditions
-            )
-            is not None
-        )
-
-    def test_get_event_conditions(
-        self,
-        logical_condition: LogicalCondition,
-        event_conditions: list[RealCondition],
-    ):
-        assert logical_condition.get_event_conditions() == event_conditions
-
-    def test_get_type(
-        self,
-        logical_condition: LogicalCondition,
-        logical_condition_type: LogicalCondition.Type,
-    ):
-        assert logical_condition.get_type() == logical_condition_type
-
-    @pytest.mark.parametrize(
-        "logical_condition_type,expected_result",
-        (
-            (LogicalCondition.Type.And, False),
-            (LogicalCondition.Type.Or, True),
-        ),
-    )
-    def test_is_satisfied(
-        self,
-        logical_condition_type: LogicalCondition.Type,
-        expected_result: bool,
-        event_conditions: list[RealCondition],
-        state: State,
-    ):
-        logical_condition: LogicalCondition = LogicalCondition(
-            "Logical Condition", logical_condition_type, event_conditions
-        )
-
-        assert (
-            logical_condition.is_satisfied(previous_state=state, current_state=state)
-            == expected_result
-        )

ostk/astrodynamics/test/event_condition/test_real_condition.py

@@ -1,53 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.physics.time import Instant, DateTime, Scale
-from ostk.physics.coordinate import Frame, Position, Velocity
-
-from ostk.astrodynamics.trajectory import State
-
-from ostk.astrodynamics.event_condition import RealCondition
-
-
-@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 evaluator():
-    return lambda state: state.get_coordinates()[0]
-
-
-@pytest.fixture
-def target() -> float:
-    return 10.0
-
-
-@pytest.fixture
-def event_condition(evaluator, target: float) -> RealCondition:
-    return RealCondition(
-        "My Condition", RealCondition.Criterion.PositiveCrossing, evaluator, target
-    )
-
-
-class TestRealCondition:
-    def test_get_target(self, event_condition: RealCondition, target: float):
-        assert event_condition.get_target() == target
-
-    def test_evaluate(self, state: State, event_condition: RealCondition):
-        assert event_condition.evaluate(state) is not None
-
-    def test_is_satisfied(self, state: State, event_condition: RealCondition):
-        assert (
-            event_condition.is_satisfied(previous_state=state, current_state=state)
-            is not None
-        )

ostk/astrodynamics/test/flight/__init__.py

@@ -1 +0,0 @@
-# Apache License 2.0

ostk/astrodynamics/test/flight/profile/__init__.py

@@ -1 +0,0 @@
-# Apache License 2.0

ostk/astrodynamics/test/flight/profile/test_state.py

@@ -1,144 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-import numpy as np
-
-import ostk.mathematics as mathematics
-
-import ostk.physics as physics
-
-import ostk.astrodynamics as astrodynamics
-
-Quaternion = mathematics.geometry.d3.transformations.rotations.Quaternion
-DateTime = physics.time.DateTime
-Instant = physics.time.Instant
-Scale = physics.time.Scale
-Position = physics.coordinate.Position
-Velocity = physics.coordinate.Velocity
-Frame = physics.coordinate.Frame
-State = astrodynamics.flight.profile.State
-
-
-@pytest.fixture
-def instant() -> Instant:
-    return Instant.date_time(DateTime(2020, 1, 3, 0, 0, 0), Scale.UTC)
-
-
-@pytest.fixture
-def frame() -> Frame:
-    return Frame.GCRF()
-
-
-@pytest.fixture
-def position(frame: Frame) -> Position:
-    return Position.meters([0.0, 3.2, 3000.0], frame)
-
-
-@pytest.fixture
-def velocity(frame: Frame) -> Velocity:
-    return Velocity.meters_per_second([1.0, 3.0, 0.0], frame)
-
-
-@pytest.fixture
-def quaternion() -> Quaternion:
-    return Quaternion([0.0, 0.0, 0.0, 1.0], Quaternion.Format.XYZS)
-
-
-@pytest.fixture
-def angular_velocity() -> np.array:
-    return np.array((0.01, 0.01, 0.0))
-
-
-@pytest.fixture
-def state(
-    instant: Instant,
-    position: Position,
-    velocity: Velocity,
-    quaternion: Quaternion,
-    angular_velocity: np.array,
-    frame: Frame,
-) -> State:
-    return State(
-        instant,
-        position,
-        velocity,
-        quaternion,
-        angular_velocity,
-        frame,
-    )
-
-
-class TestState:
-    def test_constructors_list(
-        self, instant: Instant, quaternion: Quaternion, frame: Frame
-    ):
-        position = Position.meters([0.0, 3.2, 3000.0], frame)
-        velocity = Velocity.meters_per_second([1.0, 3.0, 0.0], frame)
-        angular_velocity = [0.01, 0.01, 0.0]
-
-        assert isinstance(
-            State(instant, position, velocity, quaternion, angular_velocity, frame),
-            State,
-        )
-
-    def test_constructors_tuples(
-        self, instant: Instant, quaternion: Quaternion, frame: Frame
-    ):
-        position = Position.meters((0.0, 3.2, 3000.0), frame)
-        velocity = Velocity.meters_per_second((1.0, 3.0, 0.0), frame)
-        angular_velocity = (0.01, 0.01, 0.0)
-
-        assert isinstance(
-            State(instant, position, velocity, quaternion, angular_velocity, frame),
-            State,
-        )
-
-    def test_constructors_array(
-        self, instant: Instant, quaternion: Quaternion, frame: Frame
-    ):
-        position = Position.meters(np.array((0.0, 3.2, 3000.0)), Frame.GCRF())
-        velocity = Velocity.meters_per_second(np.array((1.0, 3.0, 0.0)), Frame.GCRF())
-        angular_velocity = np.array((0.01, 0.01, 0.0))
-
-        assert isinstance(
-            State(instant, position, velocity, quaternion, angular_velocity, frame),
-            State,
-        )
-
-    def test_comparators(self, state: State):
-        assert (state == state) is True
-        assert (state != state) is False
-
-    def test_get_instant(self, instant: Instant, state: State):
-        assert isinstance(state.get_instant(), Instant)
-        assert state.get_instant() == instant
-
-    def test_get_position(self, position: Position, state: State):
-        assert isinstance(state.get_position(), Position)
-        assert state.get_position() == position
-
-    def test_get_velocity(self, velocity: Velocity, state: State):
-        assert isinstance(state.get_velocity(), Velocity)
-        assert state.get_velocity() == velocity
-
-    def test_get_attitude(self, quaternion: Quaternion, state: State):
-        assert isinstance(state.get_attitude(), Quaternion)
-        assert state.get_attitude() == quaternion
-
-    def test_get_angular_velocity(self, angular_velocity: np.array, state: State):
-        assert isinstance(state.get_angular_velocity(), np.ndarray)
-        assert np.array_equal(state.get_angular_velocity(), angular_velocity)
-
-    def test_get_frame(self, frame: Frame, state: State):
-        assert isinstance(state.get_frame(), Frame)
-        assert state.get_frame() == frame
-
-    def test_in_frame(self, frame: Frame, state: State):
-        assert isinstance(state.in_frame(frame), State)
-
-    def test_undefined(self):
-        state: State = State.undefined()
-
-        assert isinstance(state, State)
-        assert state.is_defined() is False

ostk/astrodynamics/test/flight/system/__init__.py

@@ -1 +0,0 @@
-# Apache License 2.0

ostk/astrodynamics/test/flight/system/dynamics/__init__.py

@@ -1 +0,0 @@
-# Apache License 2.0

ostk/astrodynamics/test/flight/system/dynamics/test_atmospheric_drag.py

@@ -1,89 +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.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.atmospheric import Earth as EarthAtmosphericModel
-from ostk.physics.environment.gravitational import Earth as EarthGravitationalModel
-from ostk.physics.environment.magnetic import Earth as EarthMagneticModel
-from ostk.physics.environment.objects.celestial_bodies import Earth
-
-from ostk.astrodynamics.trajectory import State
-from ostk.astrodynamics.flight.system import SatelliteSystem
-from ostk.astrodynamics.flight.system import Dynamics
-from ostk.astrodynamics.flight.system.dynamics import AtmosphericDrag
-
-
-@pytest.fixture
-def earth() -> Earth:
-    return Earth.from_models(
-        EarthGravitationalModel(EarthGravitationalModel.Type.Undefined),
-        EarthMagneticModel(EarthMagneticModel.Type.Undefined),
-        EarthAtmosphericModel(EarthAtmosphericModel.Type.Exponential),
-    )
-
-
-@pytest.fixture
-def satellite_system() -> SatelliteSystem:
-    mass = Mass(100.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.ndarray(shape=(3, 3))
-    surface_area = 500.0
-    drag_coefficient = 2.1
-
-    return SatelliteSystem(
-        mass, satellite_geometry, inertia_tensor, surface_area, drag_coefficient
-    )
-
-
-@pytest.fixture
-def dynamics(earth: Earth, satellite_system: SatelliteSystem) -> AtmosphericDrag:
-    return AtmosphericDrag(earth, satellite_system)
-
-
-@pytest.fixture
-def state() -> State:
-    frame: Frame = Frame.GCRF()
-    position: Position = Position.meters([7000000.0, 0.0, 0.0], frame)
-    velocity: Velocity = Velocity.meters_per_second([0.0, 7546.05329, 0.0], frame)
-    instant = Instant.date_time(DateTime(2021, 3, 20, 12, 0, 0), Scale.UTC)
-    return State(instant, position, velocity)
-
-
-class TestAtmosphericDrag:
-    def test_constructors(self, dynamics: AtmosphericDrag):
-        assert dynamics is not None
-        assert isinstance(dynamics, AtmosphericDrag)
-        assert isinstance(dynamics, Dynamics)
-        assert dynamics.is_defined()
-
-    def test_getters(self, dynamics: AtmosphericDrag, earth: Earth):
-        assert dynamics.get_celestial() == earth
-
-    def test_compute_contribution(self, dynamics: AtmosphericDrag, state: State):
-        contribution = dynamics.compute_contribution(
-            state.get_instant(), state.get_coordinates(), state.get_frame()
-        )
-
-        assert len(contribution) == 3
-        assert contribution == pytest.approx(
-            [0.0, -0.0000278707803890, -0.0000000000197640], abs=5e-11
-        )

ostk/astrodynamics/test/flight/system/dynamics/test_central_body_gravity.py

@@ -1,58 +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.physics.environment.objects.celestial_bodies import Earth
-
-from ostk.astrodynamics.trajectory import State
-from ostk.astrodynamics.flight.system import Dynamics
-from ostk.astrodynamics.flight.system.dynamics import CentralBodyGravity
-
-
-@pytest.fixture
-def earth() -> Earth:
-    return Earth.spherical()
-
-
-@pytest.fixture
-def dynamics(earth: Earth) -> CentralBodyGravity:
-    return CentralBodyGravity(earth)
-
-
-@pytest.fixture
-def state() -> State:
-    frame: Frame = Frame.GCRF()
-    position: Position = Position.meters([7000000.0, 0.0, 0.0], frame)
-    velocity: Velocity = Velocity.meters_per_second([0.0, 0.0, 0.0], frame)
-    instant = Instant.date_time(DateTime(2021, 3, 20, 12, 0, 0), Scale.UTC)
-    return State(instant, position, velocity)
-
-
-class TestCentralBodyGravity:
-    def test_constructors(
-        self,
-        dynamics: CentralBodyGravity,
-    ):
-        assert dynamics is not None
-        assert isinstance(dynamics, CentralBodyGravity)
-        assert isinstance(dynamics, Dynamics)
-        assert dynamics.is_defined()
-
-    def test_getters(self, dynamics: CentralBodyGravity, earth: Earth):
-        assert dynamics.get_celestial() == earth
-
-    def test_compute_contribution(self, dynamics: CentralBodyGravity, state: State):
-        contribution = dynamics.compute_contribution(
-            state.get_instant(), state.get_coordinates(), state.get_frame()
-        )
-
-        assert len(contribution) == 3
-        assert contribution == pytest.approx([-8.134702887755102, 0.0, 0.0])

ostk/astrodynamics/test/flight/system/dynamics/test_dynamics.py

@@ -1,37 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.astrodynamics.flight.system import Dynamics
-from ostk.astrodynamics.trajectory.state.coordinates_subset import CartesianPosition
-
-
-@pytest.fixture
-def name() -> str:
-    return "MyDynamics"
-
-
-@pytest.fixture
-def dynamics(name: str) -> Dynamics:
-    class MyDynamics(Dynamics):
-        def is_defined():
-            return True
-
-        def get_read_coordinates_subsets(self):
-            return [CartesianPosition.default()]
-
-        def get_write_coordinates_subsets(self):
-            return [CartesianPosition.default()]
-
-        def compute_contribution(self, instant, state_vector, frame):
-            return state_vector
-
-    return MyDynamics(name)
-
-
-class TestDynamics:
-    def test_subclass(self, dynamics: Dynamics):
-        assert dynamics is not None
-
-    def test_get_name(self, dynamics: Dynamics, name: str):
-        assert dynamics.get_name() == name

ostk/astrodynamics/test/flight/system/dynamics/test_position_derivative.py

@@ -1,51 +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.physics.environment.objects.celestial_bodies import Earth
-
-from ostk.astrodynamics.trajectory import State
-from ostk.astrodynamics.flight.system import Dynamics
-from ostk.astrodynamics.flight.system.dynamics import PositionDerivative
-
-
-@pytest.fixture
-def dynamics() -> PositionDerivative:
-    return PositionDerivative()
-
-
-@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.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-    return State(instant, position, velocity)
-
-
-class TestPositionDerivative:
-    def test_constructors(
-        self,
-        dynamics: PositionDerivative,
-    ):
-        assert dynamics is not None
-        assert isinstance(dynamics, PositionDerivative)
-        assert isinstance(dynamics, Dynamics)
-        assert dynamics.is_defined()
-
-    def test_compute_contribution(self, dynamics: PositionDerivative, state: State):
-        contribution = dynamics.compute_contribution(
-            state.get_instant(), state.get_coordinates(), state.get_frame()
-        )
-        assert len(contribution) == 3
-        assert contribution == pytest.approx(state.get_coordinates()[:3])

ostk/astrodynamics/test/flight/system/dynamics/test_third_body_gravity.py

@@ -1,67 +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.physics.environment.objects.celestial_bodies import Moon
-
-from ostk.astrodynamics.trajectory import State
-from ostk.astrodynamics.flight.system import Dynamics
-from ostk.astrodynamics.flight.system.dynamics import ThirdBodyGravity
-
-
-@pytest.fixture
-def moon() -> Moon:
-    return Moon.spherical()
-
-
-@pytest.fixture
-def dynamics(moon: Moon) -> ThirdBodyGravity:
-    return ThirdBodyGravity(moon)
-
-
-@pytest.fixture
-def state() -> State:
-    frame: Frame = Frame.GCRF()
-    position: Position = Position.meters([7000000.0, 0.0, 0.0], frame)
-    velocity: Velocity = Velocity.meters_per_second([0.0, 0.0, 0.0], frame)
-    instant = Instant.date_time(DateTime(2021, 3, 20, 12, 0, 0), Scale.UTC)
-    return State(instant, position, velocity)
-
-
-class TestThirdBodyGravity:
-    def test_constructors(
-        self,
-        dynamics: ThirdBodyGravity,
-    ):
-        assert dynamics is not None
-        assert isinstance(dynamics, ThirdBodyGravity)
-        assert isinstance(dynamics, Dynamics)
-        assert dynamics.is_defined()
-
-    def test_getters(
-        self,
-        dynamics: ThirdBodyGravity,
-        moon: Moon,
-    ):
-        assert dynamics.get_celestial() == moon
-
-    def test_compute_contribution(
-        self,
-        dynamics: ThirdBodyGravity,
-        state: State,
-    ):
-        contribution = dynamics.compute_contribution(
-            state.get_instant(), state.get_coordinates(), state.get_frame()
-        )
-        assert len(contribution) == 3
-        assert contribution == pytest.approx(
-            [-4.620543790697659e-07, 2.948717888154649e-07, 1.301648617451192e-07]
-        )