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

ostk/astrodynamics/test/trajectory/test_local_orbital_frame_factory.py

@@ -1,64 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-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
-
-
-@pytest.fixture
-def local_orbital_frame_factory() -> LocalOrbitalFrameFactory:
-    return LocalOrbitalFrameFactory.VNC(Frame.GCRF())
-
-
-@pytest.fixture
-def instant() -> Instant:
-    return Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-
-
-@pytest.fixture
-def position_vector() -> list:
-    return [7500000.0, 0.0, 0.0]
-
-
-@pytest.fixture
-def velocity_vector() -> list:
-    return [0.0, 5335.865450622126, 5335.865450622126]
-
-
-class TestLocalOrbitalFrameFactory:
-    def test_constructors(self):
-        assert LocalOrbitalFrameFactory.VNC(Frame.GCRF()) is not None
-        assert LocalOrbitalFrameFactory.NED(Frame.GCRF()) is not None
-        assert LocalOrbitalFrameFactory.LVLH(Frame.GCRF()) is not None
-        assert LocalOrbitalFrameFactory.QSW(Frame.GCRF()) is not None
-        assert LocalOrbitalFrameFactory.TNW(Frame.GCRF()) is not None
-        assert LocalOrbitalFrameFactory.VVLH(Frame.GCRF()) is not None
-
-    def test_generate_frame(
-        self,
-        local_orbital_frame_factory: LocalOrbitalFrameFactory,
-        instant: Instant,
-        position_vector: list,
-        velocity_vector: list,
-    ):
-        frame = local_orbital_frame_factory.generate_frame(
-            instant,
-            position_vector,
-            velocity_vector,
-        )
-
-        assert frame is not None
-        assert frame.is_defined()
-
-    def test_is_defined(
-        self,
-        local_orbital_frame_factory: LocalOrbitalFrameFactory,
-    ):
-        assert local_orbital_frame_factory.is_defined()

ostk/astrodynamics/test/trajectory/test_model.py

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

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,387 +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, Sun
-from ostk.physics.environment.gravitational import Earth as EarthGravitationalModel
-from ostk.physics.environment.magnetic import Earth as EarthMagneticModel
-from ostk.physics.environment.atmospheric import Earth as EarthAtmosphericModel
-
-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.state import NumericalSolver
-
-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 import CentralBodyGravity
-from ostk.astrodynamics.flight.system.dynamics.thruster import ConstantThrust
-from ostk.astrodynamics.flight.system.dynamics import PositionDerivative
-from ostk.astrodynamics.flight.system.dynamics import AtmosphericDrag
-from ostk.astrodynamics.trajectory import State
-from ostk.astrodynamics.trajectory.state import CoordinatesSubset, CoordinatesBroker
-from ostk.astrodynamics.trajectory.state.coordinates_subset import (
-    CartesianPosition,
-    CartesianVelocity,
-)
-from ostk.astrodynamics.trajectory import Propagator
-
-from ostk.astrodynamics.event_condition import InstantCondition
-
-
-@pytest.fixture
-def propulsion_system() -> PropulsionSystem:
-    return PropulsionSystem(
-        1.0,
-        150.0,
-    )
-
-
-@pytest.fixture
-def satellite_system(propulsion_system: PropulsionSystem) -> 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,
-        propulsion_system,
-    )
-
-
-@pytest.fixture
-def earth() -> Earth:
-    return Earth.from_models(
-        EarthGravitationalModel(EarthGravitationalModel.Type.EGM96),
-        EarthMagneticModel(EarthMagneticModel.Type.Undefined),
-        EarthAtmosphericModel(EarthAtmosphericModel.Type.Exponential),
-    )
-
-
-@pytest.fixture
-def environment(earth) -> Environment:
-    sun = Sun.default()
-
-    return Environment(Instant.J2000(), [earth, sun])
-
-
-@pytest.fixture
-def coordinates_broker():
-    return CoordinatesBroker(
-        [
-            CartesianPosition.default(),
-            CartesianVelocity.default(),
-            CoordinatesSubset.mass(),
-            CoordinatesSubset.surface_area(),
-            CoordinatesSubset.drag_coefficient(),
-        ]
-    )
-
-
-@pytest.fixture
-def state(
-    satellite_system: SatelliteSystem, coordinates_broker: CoordinatesBroker
-) -> State:
-    instant: Instant = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-
-    propellant_mass: float = 10.0
-
-    coordinates: list = [
-        7500000.0,
-        0.0,
-        0.0,
-        0.0,
-        5335.865450622126,
-        5335.865450622126,
-        satellite_system.get_mass().in_kilograms() + propellant_mass,
-    ]
-
-    return State(instant, coordinates, Frame.GCRF(), coordinates_broker)
-
-
-@pytest.fixture
-def state_low_altitude(
-    satellite_system: SatelliteSystem, coordinates_broker: CoordinatesBroker
-) -> State:
-    instant: Instant = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-
-    propellant_mass: float = 10.0
-    area: float = satellite_system.get_cross_sectional_surface_area()
-    cd: float = satellite_system.get_drag_coefficient()
-
-    coordinates: list = [
-        7000000.0,
-        0.0,
-        0.0,
-        0.0,
-        5335.865450622126,
-        5335.865450622126,
-        satellite_system.get_mass().in_kilograms() + propellant_mass,
-        area,
-        cd,
-    ]
-
-    return State(instant, coordinates, Frame.GCRF(), coordinates_broker)
-
-
-@pytest.fixture
-def central_body_gravity() -> CentralBodyGravity:
-    return CentralBodyGravity(Earth.WGS84(20, 0))
-
-
-@pytest.fixture
-def atmospheric_drag(environment, satellite_system) -> AtmosphericDrag:
-    return AtmosphericDrag(environment.access_celestial_object_with_name("Earth"))
-
-
-@pytest.fixture
-def position_derivative() -> PositionDerivative:
-    return PositionDerivative()
-
-
-@pytest.fixture
-def local_orbital_frame_direction() -> LocalOrbitalFrameDirection:
-    return LocalOrbitalFrameDirection(
-        [1.0, 0.0, 0.0],
-        LocalOrbitalFrameFactory.VNC(Frame.GCRF()),
-    )
-
-
-@pytest.fixture
-def constant_thrust(
-    satellite_system: SatelliteSystem,
-    local_orbital_frame_direction: LocalOrbitalFrameDirection,
-) -> ConstantThrust:
-    return ConstantThrust(satellite_system, local_orbital_frame_direction)
-
-
-@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_calculate_states_at_with_drag(
-        self,
-        numerical_solver: NumericalSolver,
-        dynamics: list[Dynamics],
-        atmospheric_drag: AtmosphericDrag,
-        state_low_altitude: State,
-    ):
-        propagator: Propagator = Propagator(
-            numerical_solver, dynamics + [atmospheric_drag]
-        )
-
-        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),
-            Instant.date_time(DateTime(2018, 1, 1, 0, 30, 0), Scale.UTC),
-            Instant.date_time(DateTime(2018, 1, 1, 0, 40, 0), Scale.UTC),
-        ]
-
-        _ = propagator.calculate_states_at(state_low_altitude, instant_array)
-
-    def test_calculate_states_at_with_thrust(
-        self,
-        numerical_solver: NumericalSolver,
-        dynamics: list[Dynamics],
-        constant_thrust: ConstantThrust,
-        state: State,
-    ):
-        propagator: Propagator = Propagator(
-            numerical_solver, dynamics + [constant_thrust]
-        )
-
-        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),
-            Instant.date_time(DateTime(2018, 1, 1, 0, 30, 0), Scale.UTC),
-            Instant.date_time(DateTime(2018, 1, 1, 0, 40, 0), Scale.UTC),
-        ]
-
-        _ = 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,134 +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,
-        coordinates_broker: CoordinatesBroker,
-    ):
-        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()
-        assert state.get_coordinates_subsets() == coordinates_broker.get_subsets()
-
-    def test_in_frame(
-        self,
-        state: State,
-        frame: Frame,
-    ):
-        assert state.in_frame(frame) == state
-        assert state.in_frame(Frame.ITRF()) != state
-
-    def test_extract_coordinates(
-        self,
-        state: State,
-    ):
-        position_coordinates = state.extract_coordinates(CartesianPosition.default())
-        velocity_coordinates = state.extract_coordinates(CartesianVelocity.default())
-        len(position_coordinates) == 3
-        len(velocity_coordinates) == 3
-        assert (position_coordinates == state.get_position().get_coordinates()).all()
-        assert (velocity_coordinates == state.get_velocity().get_coordinates()).all()