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

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_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()