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

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

@@ -1,55 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-import ostk.physics as physics
-
-import ostk.astrodynamics as astrodynamics
-
-Length = physics.units.Length
-Angle = physics.units.Angle
-Scale = physics.time.Scale
-Instant = physics.time.Instant
-Interval = physics.time.Interval
-DateTime = physics.time.DateTime
-Position = physics.coordinate.Position
-Velocity = physics.coordinate.Velocity
-Frame = physics.coordinate.Frame
-Environment = physics.Environment
-
-Trajectory = astrodynamics.Trajectory
-Model = astrodynamics.trajectory.Model
-Orbit = astrodynamics.trajectory.Orbit
-Pass = astrodynamics.trajectory.orbit.Pass
-Kepler = astrodynamics.trajectory.orbit.models.Kepler
-COE = astrodynamics.trajectory.orbit.models.kepler.COE
-SGP4 = astrodynamics.trajectory.orbit.models.SGP4
-TLE = astrodynamics.trajectory.orbit.models.sgp4.TLE
-State = astrodynamics.trajectory.State
-Access = astrodynamics.Access
-
-earth = Environment.default().access_celestial_object_with_name("Earth")
-
-
-def test_trajectory_orbit_pass():
-    pass_type = Pass.Type.Partial
-    pass_revolution_number = 123
-    pass_start_instant = Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
-    pass_end_instant = Instant.date_time(DateTime(2018, 1, 1, 1, 0, 0), Scale.UTC)
-    pass_interval = Interval.closed(pass_start_instant, pass_end_instant)
-
-    pass_ = Pass(pass_type, pass_revolution_number, pass_interval)
-
-    assert pass_ is not None
-    assert isinstance(pass_, Pass)
-
-    assert pass_.is_defined()
-    assert pass_.is_complete() is not None
-    assert pass_.get_type() is not None
-    assert pass_.get_revolution_number() is not None
-    # Interval conversion to Python type of issue
-    # assert pass_.get_interval() is not None
-
-    assert Pass.string_from_type(Pass.Type.Complete) is not None
-    assert Pass.string_from_phase(Pass.Phase.Ascending) is not None
-    assert Pass.string_from_quarter(Pass.Quarter.First) is not None

ostk/astrodynamics/test/trajectory/state/coordinates_subset/test_cartesian_position.py

@@ -1,103 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.physics.time import Instant
-from ostk.physics.coordinate import Frame
-
-from ostk.astrodynamics.trajectory.state import CoordinatesBroker, CoordinatesSubset
-from ostk.astrodynamics.trajectory.state.coordinates_subset import CartesianPosition
-
-
-@pytest.fixture
-def name() -> str:
-    return "Position"
-
-
-@pytest.fixture
-def cartesian_position() -> CartesianPosition:
-    return CartesianPosition.default()
-
-
-@pytest.fixture
-def coordinates_subsets(cartesian_position: CartesianPosition) -> list[CoordinatesSubset]:
-    return [cartesian_position]
-
-
-@pytest.fixture
-def instant() -> Instant:
-    return Instant.J2000()
-
-
-@pytest.fixture
-def frame() -> Frame:
-    return Frame.GCRF()
-
-
-@pytest.fixture
-def coordinates_broker(coordinates_subsets: list[CoordinatesSubset]) -> CoordinatesBroker:
-    return CoordinatesBroker(coordinates_subsets)
-
-
-@pytest.fixture
-def coordinates() -> list[float]:
-    return [7000000.0, 0.0, 0.0]
-
-
-@pytest.fixture
-def another_coordinates() -> list[float]:
-    return [0.0, 7000000.0, 0.0]
-
-
-class TestCoordinatesSubset:
-    def test_constructor(self, name: str):
-        assert CartesianPosition(name) is not None
-
-    def test_add(
-        self,
-        cartesian_position: CartesianPosition,
-        instant: Instant,
-        frame: Frame,
-        coordinates: list[float],
-        another_coordinates: list[float],
-        coordinates_broker: CoordinatesBroker,
-    ):
-        assert all(
-            cartesian_position.add(
-                instant, coordinates, another_coordinates, frame, coordinates_broker
-            )
-            == [7000000.0, 7000000.0, 0.0]
-        )
-
-    def test_subtract(
-        self,
-        cartesian_position: CartesianPosition,
-        instant: Instant,
-        frame: Frame,
-        coordinates: list[float],
-        another_coordinates: list[float],
-        coordinates_broker: CoordinatesBroker,
-    ):
-        assert all(
-            cartesian_position.subtract(
-                instant, coordinates, another_coordinates, frame, coordinates_broker
-            )
-            == [7000000.0, -7000000.0, 0.0]
-        )
-
-    def test_in_frame(
-        self,
-        cartesian_position: CartesianPosition,
-        instant: Instant,
-        frame: Frame,
-        coordinates: list[float],
-        another_coordinates: list[float],
-        coordinates_broker: CoordinatesBroker,
-    ):
-        for value, expected in zip(
-            cartesian_position.in_frame(
-                instant, coordinates, frame, Frame.ITRF(), coordinates_broker
-            ),
-            [1238864.12746338, 6889500.39136482, -176.262107699686],
-        ):
-            assert value == pytest.approx(expected, rel=1e-14)

ostk/astrodynamics/test/trajectory/state/coordinates_subset/test_cartesian_velocity.py

@@ -1,113 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.physics.time import Instant
-from ostk.physics.coordinate import Frame
-
-from ostk.astrodynamics.trajectory.state import CoordinatesBroker, CoordinatesSubset
-from ostk.astrodynamics.trajectory.state.coordinates_subset import (
-    CartesianVelocity,
-    CartesianPosition,
-)
-
-
-@pytest.fixture
-def name() -> str:
-    return "Position"
-
-
-@pytest.fixture
-def cartesian_position() -> CartesianPosition:
-    return CartesianPosition.default()
-
-
-@pytest.fixture
-def cartesian_velocity() -> CartesianVelocity:
-    return CartesianVelocity.default()
-
-
-@pytest.fixture
-def coordinates_subsets(
-    cartesian_position: CartesianPosition, cartesian_velocity: CartesianVelocity
-) -> list[CoordinatesSubset]:
-    return [cartesian_position, cartesian_velocity]
-
-
-@pytest.fixture
-def instant() -> Instant:
-    return Instant.J2000()
-
-
-@pytest.fixture
-def frame() -> Frame:
-    return Frame.GCRF()
-
-
-@pytest.fixture
-def coordinates_broker(coordinates_subsets: list[CoordinatesSubset]) -> CoordinatesBroker:
-    return CoordinatesBroker(coordinates_subsets)
-
-
-@pytest.fixture
-def coordinates() -> list[float]:
-    return [7000000.0, 0.0, 0.0, 0.0, 5335.865450622126, 5335.865450622126]
-
-
-@pytest.fixture
-def another_coordinates() -> list[float]:
-    return [0.0, 7000000.0, 0.0, 5335.865450622126, 0.0, 0.0]
-
-
-class TestCoordinatesSubset:
-    def test_constructor(self, name: str):
-        assert CartesianVelocity(CartesianPosition.default(), name) is not None
-
-    def test_add(
-        self,
-        cartesian_velocity: CartesianVelocity,
-        instant: Instant,
-        frame: Frame,
-        coordinates: list[float],
-        another_coordinates: list[float],
-        coordinates_broker: CoordinatesBroker,
-    ):
-        assert all(
-            cartesian_velocity.add(
-                instant, coordinates, another_coordinates, frame, coordinates_broker
-            )
-            == [5335.865450622126, 5335.865450622126, 5335.865450622126]
-        )
-
-    def test_subtract(
-        self,
-        cartesian_velocity: CartesianVelocity,
-        instant: Instant,
-        frame: Frame,
-        coordinates: list[float],
-        another_coordinates: list[float],
-        coordinates_broker: CoordinatesBroker,
-    ):
-        assert all(
-            cartesian_velocity.subtract(
-                instant, coordinates, another_coordinates, frame, coordinates_broker
-            )
-            == [-5335.865450622126, 5335.865450622126, 5335.865450622126]
-        )
-
-    def test_in_frame(
-        self,
-        cartesian_velocity: CartesianVelocity,
-        instant: Instant,
-        frame: Frame,
-        coordinates: list[float],
-        another_coordinates: list[float],
-        coordinates_broker: CoordinatesBroker,
-    ):
-        for value, expected in zip(
-            cartesian_velocity.in_frame(
-                instant, coordinates, frame, Frame.ITRF(), coordinates_broker
-            ),
-            [-4749.36551256577, 854.163395375881, 5335.71857543495],
-        ):
-            assert value == pytest.approx(expected, rel=1e-14)

ostk/astrodynamics/test/trajectory/state/test_coordinates_broker.py

@@ -1,71 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.astrodynamics.trajectory.state import CoordinatesBroker, CoordinatesSubset
-
-
-@pytest.fixture
-def coordinates_subsets() -> list:
-    return [CoordinatesSubset("Subset1", 2), CoordinatesSubset("Subset2", 3)]
-
-
-@pytest.fixture
-def coordinates_broker(coordinates_subsets) -> CoordinatesBroker:
-    return CoordinatesBroker(coordinates_subsets)
-
-
-@pytest.fixture
-def coordinates() -> list[float]:
-    return [1.0, 2.0, 3.0, 4.0, 5.0]
-
-
-class TestCoordinatesBroker:
-    def test_constructor(self, coordinates_broker: CoordinatesBroker):
-        assert coordinates_broker is not None
-
-    def test_eq(self, coordinates_broker: CoordinatesBroker):
-        assert coordinates_broker == coordinates_broker
-
-    def test_ne(self, coordinates_broker: CoordinatesBroker):
-        assert (coordinates_broker != coordinates_broker) == False
-
-    def test_access_subsets(
-        self, coordinates_broker: CoordinatesBroker, coordinates_subsets: list
-    ):
-        assert coordinates_broker.access_subsets() == coordinates_subsets
-
-    def test_get_number_of_coordinates(self, coordinates_broker: CoordinatesBroker):
-        assert coordinates_broker.get_number_of_coordinates() == 5
-
-    def test_get_number_of_subsets(self, coordinates_broker: CoordinatesBroker):
-        assert coordinates_broker.get_number_of_subsets() == 2
-
-    def test_get_subsets(
-        self, coordinates_broker: CoordinatesBroker, coordinates_subsets: list
-    ):
-        assert coordinates_broker.get_subsets() == coordinates_subsets
-
-    def test_add_subset(self, coordinates_broker: CoordinatesBroker):
-        new_subset = CoordinatesSubset("NewSubset", 4)
-        number_of_coordinates = coordinates_broker.get_number_of_coordinates()
-        assert coordinates_broker.add_subset(new_subset) == number_of_coordinates
-        assert coordinates_broker.has_subset(new_subset)
-
-    def test_has_subset(
-        self, coordinates_broker: CoordinatesBroker, coordinates_subsets: list
-    ):
-        assert coordinates_broker.has_subset(coordinates_subsets[0])
-
-    def extract_coordinates(
-        self,
-        coordinates_broker: CoordinatesBroker,
-        coordinates: list[float],
-        coordinates_subsets: list[CoordinatesSubset],
-    ):
-        assert coordinates_broker.extract_coordinates(
-            coordinates, coordinates_subsets[0]
-        ) == [1.0, 2.0]
-        assert coordinates_broker.extract_coordinates(
-            coordinates, coordinates_subsets[1]
-        ) == [3.0, 4.0, 5.0]

ostk/astrodynamics/test/trajectory/state/test_coordinates_subset.py

@@ -1,43 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.astrodynamics.trajectory.state import CoordinatesSubset
-
-
-@pytest.fixture
-def name() -> str:
-    return "MySubset"
-
-
-@pytest.fixture
-def size() -> int:
-    return 3
-
-
-@pytest.fixture
-def coordinates_subset(name: str, size: int) -> CoordinatesSubset:
-    return CoordinatesSubset(name, size)
-
-
-class TestCoordinatesSubset:
-    def test_constructor(self, coordinates_subset: CoordinatesSubset):
-        assert coordinates_subset is not None
-
-    def test_eq(self, coordinates_subset: CoordinatesSubset):
-        assert coordinates_subset == coordinates_subset
-
-    def test_ne(self, coordinates_subset: CoordinatesSubset):
-        assert (coordinates_subset != coordinates_subset) == False
-
-    def test_get_id(self, coordinates_subset: CoordinatesSubset):
-        assert coordinates_subset.get_id() is not None
-
-    def test_get_name(self, coordinates_subset: CoordinatesSubset, name: str):
-        assert coordinates_subset.get_name() == name
-
-    def test_get_size(self, coordinates_subset: CoordinatesSubset, size: int):
-        assert coordinates_subset.get_size() == size
-
-    def test_mass(self):
-        assert CoordinatesSubset.mass() is not None

ostk/astrodynamics/test/trajectory/state/test_numerical_solver.py

@@ -1,242 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-import numpy as np
-import math
-
-from ostk.physics.time import Instant, Duration
-from ostk.physics.coordinate import Frame
-
-from ostk.astrodynamics.trajectory import State
-from ostk.astrodynamics.trajectory.state import (
-    NumericalSolver,
-    CoordinatesBroker,
-    CoordinatesSubset,
-)
-from ostk.astrodynamics.event_condition import RealCondition
-
-
-def oscillator(x, dxdt, _):
-    dxdt[0] = x[1]
-    dxdt[1] = -x[0]
-    return dxdt
-
-
-def get_state_vec(time: float) -> np.ndarray:
-    return np.array([math.sin(time), math.cos(time)])
-
-
-@pytest.fixture
-def coordinates_subsets() -> list[CoordinatesSubset]:
-    return [CoordinatesSubset("Subset", 2)]
-
-
-@pytest.fixture
-def coordinates_broker(coordinates_subsets: list[CoordinatesSubset]) -> CoordinatesBroker:
-    return CoordinatesBroker(coordinates_subsets)
-
-
-@pytest.fixture
-def start_instant() -> Instant:
-    return Instant.J2000()
-
-
-@pytest.fixture
-def frame() -> Frame:
-    return Frame.GCRF()
-
-
-@pytest.fixture
-def initial_state(
-    start_instant: Instant,
-    frame: Frame,
-    coordinates_broker: CoordinatesBroker,
-) -> State:
-    return State(start_instant, get_state_vec(0.0), frame, coordinates_broker)
-
-
-@pytest.fixture
-def custom_condition() -> RealCondition:
-    return RealCondition(
-        "Custom",
-        RealCondition.Criterion.StrictlyPositive,
-        lambda state: (state.get_instant() - Instant.J2000()).in_seconds(),
-        5.0,
-    )
-
-
-@pytest.fixture
-def numerical_solver_default_inputs() -> (
-    tuple[NumericalSolver.LogType, NumericalSolver.StepperType, float, float, float]
-):
-    log_type = NumericalSolver.LogType.NoLog
-    stepper_type = NumericalSolver.StepperType.RungeKuttaCashKarp54
-    initial_time_step = 5.0
-    relative_tolerance = 1.0e-15
-    absolute_tolerance = 1.0e-15
-
-    return (
-        log_type,
-        stepper_type,
-        initial_time_step,
-        relative_tolerance,
-        absolute_tolerance,
-    )
-
-
-@pytest.fixture
-def numerical_solver(numerical_solver_default_inputs) -> NumericalSolver:
-    return NumericalSolver(*numerical_solver_default_inputs)
-
-
-@pytest.fixture
-def numerical_solver_conditional() -> NumericalSolver:
-    return NumericalSolver(
-        NumericalSolver.LogType.NoLog,
-        NumericalSolver.StepperType.RungeKuttaDopri5,
-        5.0,
-        1.0e-15,
-        1.0e-15,
-    )
-
-
-class TestNumericalSolver:
-    def test_constructors(self, numerical_solver: NumericalSolver):
-        assert numerical_solver is not None
-        assert isinstance(numerical_solver, NumericalSolver)
-        assert numerical_solver.is_defined()
-
-    def test_comparators(self, numerical_solver: NumericalSolver):
-        assert numerical_solver == numerical_solver
-        assert (numerical_solver != numerical_solver) is False
-
-    def test_get_types(
-        self,
-        numerical_solver_default_inputs: tuple[
-            NumericalSolver.LogType, NumericalSolver.StepperType, float, float, float
-        ],
-        numerical_solver: NumericalSolver,
-    ):
-        (
-            log_type,
-            stepper_type,
-            initial_time_step,
-            relative_tolerance,
-            absolute_tolerance,
-        ) = numerical_solver_default_inputs
-
-        assert numerical_solver.get_log_type() == log_type
-        assert numerical_solver.get_stepper_type() == stepper_type
-        assert numerical_solver.get_time_step() == initial_time_step
-        assert numerical_solver.get_relative_tolerance() == relative_tolerance
-        assert numerical_solver.get_absolute_tolerance() == absolute_tolerance
-        assert numerical_solver.get_root_solver() is not None
-        assert numerical_solver.get_observed_states() is not None
-
-    def test_get_string_from_types(self):
-        assert (
-            NumericalSolver.string_from_stepper_type(
-                NumericalSolver.StepperType.RungeKutta4
-            )
-            == "RungeKutta4"
-        )
-        assert (
-            NumericalSolver.string_from_stepper_type(
-                NumericalSolver.StepperType.RungeKuttaCashKarp54
-            )
-            == "RungeKuttaCashKarp54"
-        )
-        assert (
-            NumericalSolver.string_from_stepper_type(
-                NumericalSolver.StepperType.RungeKuttaFehlberg78
-            )
-            == "RungeKuttaFehlberg78"
-        )
-        assert (
-            NumericalSolver.string_from_log_type(NumericalSolver.LogType.NoLog) == "NoLog"
-        )
-        assert (
-            NumericalSolver.string_from_log_type(NumericalSolver.LogType.LogConstant)
-            == "LogConstant"
-        )
-        assert (
-            NumericalSolver.string_from_log_type(NumericalSolver.LogType.LogAdaptive)
-            == "LogAdaptive"
-        )
-
-    def test_integrate_time(
-        self,
-        initial_state: State,
-        numerical_solver: NumericalSolver,
-    ):
-        duration_seconds: float = 100.0
-        end_instant: Instant = initial_state.get_instant() + Duration.seconds(
-            duration_seconds
-        )
-
-        state_vector: np.ndarray = numerical_solver.integrate_time(
-            initial_state, end_instant, oscillator
-        ).get_coordinates()
-
-        assert 5e-9 >= abs(state_vector[0] - math.sin(duration_seconds))
-        assert 5e-9 >= abs(state_vector[1] - math.cos(duration_seconds))
-
-        end_instants: list[Instant] = [
-            initial_state.get_instant() + Duration.seconds(duration)
-            for duration in np.arange(600.0, 1000.0, 50.0)
-        ]
-        states: list[State] = numerical_solver.integrate_time(
-            initial_state, end_instants, oscillator
-        )
-
-        for state, end_instant in zip(states, end_instants):
-            state_vector: np.ndarray = state.get_coordinates()
-
-            assert 5e-9 >= abs(
-                state_vector[0]
-                - math.sin((end_instant - initial_state.get_instant()).in_seconds())
-            )
-            assert 5e-9 >= abs(
-                state_vector[1]
-                - math.cos((end_instant - initial_state.get_instant()).in_seconds())
-            )
-
-    def test_integrate_time_with_condition(
-        self,
-        initial_state: State,
-        numerical_solver_conditional: NumericalSolver,
-        custom_condition: RealCondition,
-    ):
-        end_time: float = initial_state.get_instant() + Duration.seconds(100.0)
-
-        condition_solution = numerical_solver_conditional.integrate_time(
-            initial_state, end_time, oscillator, custom_condition
-        )
-
-        assert condition_solution.condition_is_satisfied
-        assert (
-            condition_solution.iteration_count
-            < numerical_solver_conditional.get_root_solver().get_maximum_iteration_count()
-        )
-        assert condition_solution.root_solver_has_converged
-
-        state_vector = condition_solution.state.get_coordinates()
-        time = (
-            condition_solution.state.get_instant() - initial_state.get_instant()
-        ).in_seconds()
-
-        assert abs(float(time - custom_condition.get_target())) < 1e-6
-
-        assert 5e-9 >= abs(state_vector[0] - math.sin(time))
-        assert 5e-9 >= abs(state_vector[1] - math.cos(time))
-
-    def test_default(self):
-        assert NumericalSolver.default() is not None
-
-    def test_default_conditional(self):
-        assert NumericalSolver.default_conditional() is not None
-
-    def test_undefined(self):
-        assert NumericalSolver.undefined() is not None
-        assert NumericalSolver.undefined().is_defined() is False

ostk/astrodynamics/test/trajectory/test_local_orbital_frame_direction.py

@@ -1,82 +0,0 @@
-# Apache License 2.0
-
-import pytest
-
-from ostk.physics.time import Instant
-from ostk.physics.coordinate import Frame
-
-from ostk.astrodynamics.trajectory import LocalOrbitalFrameFactory
-from ostk.astrodynamics.trajectory import LocalOrbitalFrameDirection
-
-
-@pytest.fixture
-def local_orbital_frame_factory() -> LocalOrbitalFrameFactory:
-    return LocalOrbitalFrameFactory.VNC(Frame.GCRF())
-
-
-@pytest.fixture
-def direction_vector() -> list:
-    return [1.0, 0.0, 0.0]
-
-
-@pytest.fixture
-def local_orbital_frame_direction(
-    direction_vector: list,
-    local_orbital_frame_factory: LocalOrbitalFrameFactory,
-) -> LocalOrbitalFrameDirection:
-    return LocalOrbitalFrameDirection(
-        direction_vector,
-        local_orbital_frame_factory,
-    )
-
-
-class TestLocalOrbitalFrameDirection:
-    def test_constructor(
-        self,
-        direction_vector: list,
-        local_orbital_frame_factory: LocalOrbitalFrameFactory,
-    ):
-        local_orbital_frame_direction: LocalOrbitalFrameDirection = (
-            LocalOrbitalFrameDirection(
-                direction_vector,
-                local_orbital_frame_factory,
-            )
-        )
-
-        assert local_orbital_frame_direction is not None
-        assert isinstance(local_orbital_frame_direction, LocalOrbitalFrameDirection)
-
-    def test_equal_to_operator(
-        self, local_orbital_frame_direction: LocalOrbitalFrameDirection
-    ):
-        assert local_orbital_frame_direction == local_orbital_frame_direction
-        assert local_orbital_frame_direction != LocalOrbitalFrameDirection(
-            [0.0, 1.0, 0.0],
-            local_orbital_frame_direction.get_local_orbital_frame_factory(),
-        )
-
-    def test_is_defined(
-        self,
-        local_orbital_frame_direction: LocalOrbitalFrameDirection,
-    ):
-        assert local_orbital_frame_direction.is_defined()
-
-    def test_getters(
-        self,
-        local_orbital_frame_direction: LocalOrbitalFrameDirection,
-        direction_vector: list,
-        local_orbital_frame_factory: LocalOrbitalFrameFactory,
-    ):
-        assert list(local_orbital_frame_direction.get_value()) == direction_vector
-        assert (
-            local_orbital_frame_direction.get_local_orbital_frame_factory()
-            == local_orbital_frame_factory
-        )
-
-    def test_undefined(
-        self,
-    ):
-        local_orbital_frame_direction: LocalOrbitalFrameDirection = (
-            LocalOrbitalFrameDirection.undefined()
-        )
-        assert local_orbital_frame_direction.is_defined() is False