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

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]
-        )

ostk/astrodynamics/test/flight/system/dynamics/thruster/test_constant_thrust.py

@@ -1,143 +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.astrodynamics.trajectory import LocalOrbitalFrameFactory
-from ostk.astrodynamics.trajectory import LocalOrbitalFrameDirection
-
-from ostk.astrodynamics.trajectory.state import CoordinatesSubset
-from ostk.astrodynamics.trajectory.state.coordinates_subset import CartesianPosition
-from ostk.astrodynamics.trajectory.state.coordinates_subset import CartesianVelocity
-from ostk.astrodynamics.trajectory.state import CoordinatesBroker
-
-from ostk.astrodynamics.trajectory import State
-from ostk.astrodynamics.flight.system import PropulsionSystem
-from ostk.astrodynamics.flight.system import SatelliteSystem
-from ostk.astrodynamics.flight.system import Dynamics
-from ostk.astrodynamics.flight.system.dynamics.thruster import ConstantThrust
-
-
-@pytest.fixture
-def local_orbital_frame_direction() -> LocalOrbitalFrameDirection:
-    return LocalOrbitalFrameDirection(
-        [1.0, 0.0, 0.0],
-        LocalOrbitalFrameFactory.VNC(Frame.GCRF()),
-    )
-
-
-@pytest.fixture
-def propulsion_system() -> PropulsionSystem:
-    return PropulsionSystem(
-        1.0,
-        150.0,
-    )
-
-
-@pytest.fixture
-def satellite_system(propulsion_system: PropulsionSystem) -> 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 = 0.8
-    drag_coefficient = 2.1
-
-    return SatelliteSystem(
-        mass,
-        satellite_geometry,
-        inertia_tensor,
-        surface_area,
-        drag_coefficient,
-        propulsion_system,
-    )
-
-
-@pytest.fixture
-def dynamics(
-    satellite_system: SatelliteSystem,
-    local_orbital_frame_direction: LocalOrbitalFrameDirection,
-) -> ConstantThrust:
-    return ConstantThrust(
-        satellite_system,
-        local_orbital_frame_direction,
-        "Constant Thrust Thruster Dynamics",
-    )
-
-
-@pytest.fixture
-def coordinates_broker() -> CoordinatesBroker:
-    return CoordinatesBroker(
-        [
-            CartesianPosition.default(),
-            CartesianVelocity.default(),
-            CoordinatesSubset.mass(),
-        ]
-    )
-
-
-@pytest.fixture
-def state(coordinates_broker: CoordinatesBroker) -> State:
-    instant: Instant = Instant.date_time(DateTime(2021, 3, 20, 12, 0, 0), Scale.UTC)
-    coordinates: list = [7000000.0, 0.0, 0.0, 0.0, 7546.05329, 0.0, 105.0]
-
-    return State(instant, coordinates, Frame.GCRF(), coordinates_broker)
-
-
-class TestConstantThrust:
-    def test_constructors(self, dynamics: ConstantThrust):
-        assert dynamics is not None
-        assert isinstance(dynamics, ConstantThrust)
-        assert isinstance(dynamics, Dynamics)
-        assert dynamics.is_defined()
-
-    def test_compute_contribution_success(self, dynamics: ConstantThrust, state: State):
-        contribution = dynamics.compute_contribution(
-            state.get_instant(), state.get_coordinates(), state.get_frame()
-        )
-
-        assert len(contribution) == 4
-        assert contribution == pytest.approx(
-            [0.0, 0.009523809523809525, 0.0, -0.0006798108086519521], abs=5e-11
-        )
-
-    def test_compute_contribution_failure_out_of_fuel(
-        self,
-        satellite_system: SatelliteSystem,
-        coordinates_broker: CoordinatesBroker,
-        dynamics: ConstantThrust,
-    ):
-        instant: Instant = Instant.date_time(DateTime(2021, 3, 20, 12, 0, 0), Scale.UTC)
-        coordinates: list = [
-            7000000.0,
-            0.0,
-            0.0,
-            0.0,
-            7546.05329,
-            0.0,
-            satellite_system.get_mass().in_kilograms(),
-        ]
-        state = State(instant, coordinates, Frame.GCRF(), coordinates_broker)
-
-        with pytest.raises(RuntimeError):
-            contribution = dynamics.compute_contribution(
-                state.get_instant(), state.get_coordinates(), state.get_frame()
-            )

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

@@ -1,46 +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
-
-Mass = physics.units.Mass
-
-PropulsionSystem = astrodynamics.flight.system.PropulsionSystem
-
-
-@pytest.fixture
-def propulsion_system() -> PropulsionSystem:
-    return PropulsionSystem(
-        1.0,
-        150.0,
-    )
-
-
-class TestPropulsionSystem:
-    def test_constructors(
-        self,
-        propulsion_system: PropulsionSystem,
-    ):
-        assert propulsion_system is not None
-        assert isinstance(propulsion_system, PropulsionSystem)
-        assert propulsion_system.is_defined()
-
-    def test_comparators(
-        self,
-        propulsion_system: PropulsionSystem,
-    ):
-        assert (propulsion_system == propulsion_system) is True
-        assert (propulsion_system != propulsion_system) is False
-
-    def test_is_defined(
-        self,
-        propulsion_system: PropulsionSystem,
-    ):
-        assert propulsion_system.is_defined()

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

@@ -1,77 +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
-
-Cuboid = mathematics.geometry.d3.objects.Cuboid
-Composite = mathematics.geometry.d3.objects.Composite
-Point = mathematics.geometry.d3.objects.Point
-
-Mass = physics.units.Mass
-
-SatelliteSystem = astrodynamics.flight.system.SatelliteSystem
-
-
-@pytest.fixture
-def satellite_system_default_inputs():
-    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.ndarray(shape=(3, 3))
-    surface_area = 0.8
-    drag_coefficient = 2.2
-
-    return (mass, satellite_geometry, inertia_tensor, surface_area, drag_coefficient)
-
-
-@pytest.fixture
-def satellite_system(satellite_system_default_inputs) -> SatelliteSystem:
-    return SatelliteSystem(*satellite_system_default_inputs)
-
-
-class TestSatelliteSystem:
-    def test_constructors(
-        self,
-        satellite_system: SatelliteSystem,
-    ):
-        assert satellite_system is not None
-        assert isinstance(satellite_system, SatelliteSystem)
-        assert satellite_system.is_defined()
-
-    def test_comparators(
-        self,
-        satellite_system: SatelliteSystem,
-    ):
-        assert (satellite_system == satellite_system) is True
-        assert (satellite_system != satellite_system) is False
-
-    def test_getters(
-        self,
-        satellite_system_default_inputs,
-        satellite_system: SatelliteSystem,
-    ):
-        (
-            mass,
-            satellite_geometry,
-            inertia_tensor,
-            surface_area,
-            drag_coefficient,
-        ) = satellite_system_default_inputs
-
-        assert satellite_system.get_mass() == mass
-        assert satellite_system.get_geometry() == satellite_geometry
-        assert np.array_equal(satellite_system.get_inertia_tensor(), inertia_tensor)
-        assert satellite_system.get_cross_sectional_surface_area() == surface_area
-        assert satellite_system.get_drag_coefficient() == drag_coefficient

ostk/astrodynamics/test/flight/test_profile.py

@@ -1,153 +0,0 @@
-# Apache License 2.0
-
-from datetime import datetime
-
-import pytest
-
-from ostk.mathematics.geometry.d3.transformations.rotations import Quaternion
-
-from ostk.physics import Environment
-from ostk.physics.time import DateTime
-from ostk.physics.time import Time
-from ostk.physics.time import Scale
-from ostk.physics.time import Instant
-from ostk.physics.units import Length
-from ostk.physics.coordinate import Transform
-from ostk.physics.coordinate import Position
-from ostk.physics.coordinate import Velocity
-from ostk.physics.coordinate import Frame
-from ostk.physics.coordinate import Axes
-from ostk.physics.coordinate.frame.providers import Dynamic as DynamicProvider
-
-from ostk.astrodynamics import Trajectory
-from ostk.astrodynamics.trajectory import Orbit
-from ostk.astrodynamics.flight import Profile
-from ostk.astrodynamics.flight.profile import State
-from ostk.astrodynamics.flight.profile.models import Transform as TransformModel
-from ostk.astrodynamics.flight.profile.models import Tabulated as TabulatedModel
-
-
-@pytest.fixture
-def instant() -> Instant:
-    return Instant.date_time(DateTime(2020, 1, 3), Scale.UTC)
-
-
-@pytest.fixture
-def profile() -> Profile:
-    def dynamic_provider_generator(instant: Instant):
-        return Transform.identity(instant)
-
-    return Profile(
-        model=TransformModel(
-            dynamic_provider=DynamicProvider(dynamic_provider_generator),
-            frame=Frame.GCRF(),
-        ),
-    )
-
-
-class TestProfile:
-    def test_constructors(self, profile: Profile):
-        assert profile is not None
-        assert isinstance(profile, Profile)
-
-    def test_get_state_at(self, profile: Profile, instant: Instant):
-        state: State = profile.get_state_at(instant)
-
-        assert state is not None
-        assert isinstance(state, State)
-        state.is_defined()
-
-    def test_get_states_at(self, profile: Profile, instant: Instant):
-        states = profile.get_states_at([instant, instant])
-
-        assert states is not None
-
-    def test_get_axes_at(self, profile: Profile, instant: Instant):
-        axes = profile.get_axes_at(instant)
-
-        assert axes is not None
-        assert isinstance(axes, Axes)
-
-    def test_get_body_frame(self, profile: Profile, instant: Instant):
-        frame = profile.get_body_frame("Name")
-
-        assert frame is not None
-        assert isinstance(frame, Frame)
-
-    def test_undefined(self):
-        profile: Profile = Profile.undefined()
-
-        assert profile is not None
-        assert isinstance(profile, Profile)
-        assert profile.is_defined() is False
-
-    def test_inertial_pointing(self):
-        quaternion: Quaternion = Quaternion([0.0, 0.0, 0.0, 1.0], Quaternion.Format.XYZS)
-
-        trajectory: Trajectory = Trajectory.position(
-            Position.meters((0.0, 0.0, 0.0), Frame.GCRF())
-        )
-
-        profile: Profile = Profile.inertial_pointing(trajectory, quaternion)
-
-        assert profile is not None
-        assert isinstance(profile, Profile)
-        assert profile.is_defined()
-
-    def test_nadir_pointing(self):
-        environment = Environment.default()
-
-        orbit = Orbit.sun_synchronous(
-            epoch=Instant.date_time(datetime(2020, 1, 1, 0, 0, 0), Scale.UTC),
-            altitude=Length.kilometers(500.0),
-            local_time_at_descending_node=Time(14, 0, 0),
-            celestial_object=environment.access_celestial_object_with_name("Earth"),
-        )
-
-        profile: Profile = Profile.nadir_pointing(orbit, Orbit.FrameType.VVLH)
-
-        assert profile is not None
-        assert isinstance(profile, Profile)
-        assert profile.is_defined()
-
-    def test_tabulated(self):
-        profile = Profile(
-            model=TabulatedModel(
-                states=[
-                    State(
-                        instant=Instant.date_time(
-                            datetime(2020, 1, 1, 0, 0, 0), Scale.UTC
-                        ),
-                        position=Position.meters((0.0, 0.0, 0.0), Frame.GCRF()),
-                        velocity=Velocity.meters_per_second(
-                            (0.0, 0.0, 0.0), Frame.GCRF()
-                        ),
-                        attitude=Quaternion.unit(),
-                        angular_velocity=(0.0, 0.0, 0.0),
-                        reference_frame=Frame.GCRF(),
-                    ),
-                    State(
-                        instant=Instant.date_time(
-                            datetime(2020, 1, 1, 0, 1, 0), Scale.UTC
-                        ),
-                        position=Position.meters((0.0, 0.0, 0.0), Frame.GCRF()),
-                        velocity=Velocity.meters_per_second(
-                            (0.0, 0.0, 0.0), Frame.GCRF()
-                        ),
-                        attitude=Quaternion.unit(),
-                        angular_velocity=(0.0, 0.0, 0.0),
-                        reference_frame=Frame.GCRF(),
-                    ),
-                ],
-            ),
-        )
-
-        assert isinstance(profile, Profile)
-        assert profile.is_defined()
-
-        assert (
-            profile.get_state_at(
-                Instant.date_time(datetime(2020, 1, 1, 0, 0, 30), Scale.UTC)
-            )
-            is not None
-        )

ostk/astrodynamics/test/flight/test_system.py

@@ -1,55 +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
-
-Cuboid = mathematics.geometry.d3.objects.Cuboid
-Composite = mathematics.geometry.d3.objects.Composite
-Point = mathematics.geometry.d3.objects.Point
-
-Mass = physics.units.Mass
-
-System = astrodynamics.flight.System
-
-
-@pytest.fixture
-def system_default_inputs():
-    mass = Mass(90.0, Mass.Unit.Kilogram)
-    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],
-        )
-    )
-
-    return (mass, geometry)
-
-
-@pytest.fixture
-def system(system_default_inputs) -> System:
-    return System(*system_default_inputs)
-
-
-class TestSatelliteSystem:
-    def test_constructors(self, system: System):
-        assert system is not None
-        assert isinstance(system, System)
-        assert system.is_defined()
-
-    def test_comparators(self, system: System):
-        assert (system == system) is True
-        assert (system != system) is False
-
-    def test_getters(self, system_default_inputs, system: System):
-        (mass, geometry) = system_default_inputs
-
-        assert system.get_mass() == mass
-        assert system.get_geometry() == geometry

ostk/astrodynamics/test/solvers/__init__.py

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

ostk/astrodynamics/test/solvers/test_temporal_condition_solver.py

@@ -1,153 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.physics.units import Length
-from ostk.physics.units import Angle
-from ostk.physics.time import Instant
-from ostk.physics.time import Duration
-from ostk.physics.time import Interval
-from ostk.physics.time import DateTime
-from ostk.physics.time import Scale
-from ostk.physics import Environment
-
-from ostk.astrodynamics.trajectory import Orbit
-from ostk.astrodynamics.trajectory.orbit.models import Kepler
-from ostk.astrodynamics.trajectory.orbit.models.kepler import COE
-from ostk.astrodynamics.access import Generator
-from ostk.astrodynamics.solvers import TemporalConditionSolver
-
-
-@pytest.fixture
-def temporal_condition_solver() -> TemporalConditionSolver:
-    return TemporalConditionSolver(
-        time_step=Duration.seconds(30.0),
-        tolerance=Duration.milliseconds(1.0),
-        maximum_iteration_count=1234,
-    )
-
-
-@pytest.fixture
-def interval() -> Interval:
-    return Interval.closed(
-        Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC),
-        Instant.date_time(DateTime(2018, 1, 1, 2, 0, 0), Scale.UTC),
-    )
-
-
-class TestTemporalConditionSolver:
-    def test_constructor_success(self):
-        temporal_condition_solver = TemporalConditionSolver(
-            time_step=Duration.seconds(30.0),
-            tolerance=Duration.milliseconds(1.0),
-            maximum_iteration_count=1234,
-        )
-
-        assert isinstance(temporal_condition_solver, TemporalConditionSolver)
-
-    def test_getters_success(
-        self,
-        temporal_condition_solver: TemporalConditionSolver,
-    ):
-        assert temporal_condition_solver.get_time_step() == Duration.seconds(30.0)
-        assert temporal_condition_solver.get_tolerance() == Duration.milliseconds(1.0)
-        assert temporal_condition_solver.get_maximum_iteration_count() == 1234
-
-    def test_solve_success_one_condition_always_true(
-        self,
-        temporal_condition_solver: TemporalConditionSolver,
-        interval: Interval,
-    ):
-        solution: list[Interval] = temporal_condition_solver.solve(
-            condition=lambda _: True,
-            interval=interval,
-        )
-
-        assert isinstance(solution, list)
-        assert solution == [interval]
-
-    def test_solve_success_one_condition_always_false(
-        self,
-        temporal_condition_solver: TemporalConditionSolver,
-        interval: Interval,
-    ):
-        solution: list[Interval] = temporal_condition_solver.solve(
-            condition=lambda _: False,
-            interval=interval,
-        )
-
-        assert isinstance(solution, list)
-        assert solution == []
-
-    def test_solve_success_multiple_conditions_always_true(
-        self,
-        temporal_condition_solver: TemporalConditionSolver,
-        interval: Interval,
-    ):
-        solution: list[Interval] = temporal_condition_solver.solve(
-            conditions=[
-                lambda _: True,
-                lambda _: True,
-            ],
-            interval=interval,
-        )
-
-        assert isinstance(solution, list)
-        assert solution == [interval]
-
-    def test_solve_success_multiple_conditions_one_always_false(
-        self,
-        temporal_condition_solver: TemporalConditionSolver,
-        interval: Interval,
-    ):
-        solution: list[Interval] = temporal_condition_solver.solve(
-            conditions=[
-                lambda _: True,
-                lambda _: False,
-            ],
-            interval=interval,
-        )
-
-        assert isinstance(solution, list)
-        assert solution == []
-
-    def test_solve_success_using_access_generator(
-        self,
-        temporal_condition_solver: TemporalConditionSolver,
-        interval: Interval,
-    ):
-        environment = Environment.default()
-
-        generator = Generator(
-            environment=environment,
-        )
-
-        earth = environment.access_celestial_object_with_name("Earth")
-
-        trajectory = Orbit(
-            model=Kepler(
-                coe=COE(
-                    semi_major_axis=Length.kilometers(7000.0),
-                    eccentricity=0.0,
-                    inclination=Angle.degrees(45.0),
-                    raan=Angle.degrees(0.0),
-                    aop=Angle.degrees(0.0),
-                    true_anomaly=Angle.degrees(0.0),
-                ),
-                epoch=Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC),
-                celestial_object=earth,
-                perturbation_type=Kepler.PerturbationType.No,
-            ),
-            celestial_object=earth,
-        )
-
-        solution: list[Interval] = temporal_condition_solver.solve(
-            condition=generator.get_condition_function(
-                from_trajectory=trajectory,
-                to_trajectory=trajectory,
-            ),
-            interval=interval,
-        )
-
-        assert isinstance(solution, list)
-        assert solution == [interval]

ostk/astrodynamics/test/trajectory/__init__.py

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

ostk/astrodynamics/test/trajectory/orbit/__init__.py

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