open-space-toolkit-astrodynamics 5.1.5__py38-none-any.whl → 5.2.0__py38-none-any.whl

ostk/astrodynamics/test/trajectory/test_state_builder.py

@@ -0,0 +1,171 @@
+# 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 Frame
+
+from ostk.astrodynamics.trajectory import (
+    State,
+    StateBuilder,
+)
+from ostk.astrodynamics.trajectory.state import (
+    CoordinatesBroker,
+    CoordinatesSubset,
+)
+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 coordinates_subsets() -> list[CoordinatesSubset]:
+    return [CartesianPosition.default(), CartesianVelocity.default()]
+
+
+@pytest.fixture
+def coordinates() -> list[float]:
+    return [1.0, 2.0, 3.0, 4.0, 5.0, 6.0]
+
+
+@pytest.fixture
+def coordinates_broker(
+    coordinates_subsets: list[CoordinatesSubset],
+) -> CoordinatesBroker:
+    return CoordinatesBroker(coordinates_subsets)
+
+
+@pytest.fixture
+def state(
+    instant: Instant,
+    coordinates: list[float],
+    frame: Frame,
+    coordinates_broker: CoordinatesBroker,
+) -> State:
+    return State(instant, coordinates, frame, coordinates_broker)
+
+
+@pytest.fixture
+def state_builder(frame: Frame, coordinates_broker: CoordinatesBroker) -> State:
+    return StateBuilder(frame, coordinates_broker)
+
+
+class TestStateBuilder:
+    def test_broker_constructor(
+        self,
+        frame: Frame,
+        coordinates_broker: CoordinatesBroker,
+    ):
+        builder = StateBuilder(frame, coordinates_broker)
+        assert builder is not None
+        assert isinstance(builder, StateBuilder)
+        assert builder.is_defined()
+
+    def test_subsets_constructor(
+        self,
+        frame: Frame,
+        coordinates_subsets: list[CoordinatesSubset],
+    ):
+        builder = StateBuilder(frame, coordinates_subsets)
+        assert builder is not None
+        assert isinstance(builder, StateBuilder)
+        assert builder.is_defined()
+
+    def test_state_constructor(
+        self,
+        state: State,
+    ):
+        builder = StateBuilder(state)
+        assert builder is not None
+        assert isinstance(builder, StateBuilder)
+        assert builder.is_defined()
+
+    def test_comparators(self, state_builder: StateBuilder):
+        assert (state_builder == state_builder) is True
+        assert (state_builder != state_builder) is False
+
+    def test_operators(
+        self,
+        state_builder: StateBuilder,
+    ):
+        added_builder: StateBuilder = state_builder + CoordinatesSubset.mass()
+        assert isinstance(added_builder, StateBuilder)
+        assert state_builder != added_builder
+
+        subtracted_builder: StateBuilder = state_builder - CartesianPosition.default()
+        assert isinstance(subtracted_builder, StateBuilder)
+        assert state_builder != subtracted_builder
+
+    def test_build(
+        self,
+        instant: Instant,
+        state_builder: StateBuilder,
+    ):
+        coordinates = [1, 2, 3, 1, 2, 3]
+        state: State = state_builder.build(instant, coordinates)
+
+        assert state is not None
+        assert isinstance(state, State)
+        assert state.is_defined()
+        assert state.get_instant() == instant
+        assert (state.get_coordinates() == coordinates).all()
+        assert state.get_frame() == state_builder.get_frame()
+        assert state.get_coordinates_subsets() == state_builder.get_coordinates_subsets()
+
+    def test_reduce(
+        self,
+        state: State,
+    ):
+        builder = StateBuilder(state.get_frame(), [CartesianPosition.default()])
+        reduced_state: State = builder.reduce(state)
+
+        assert isinstance(reduced_state, State)
+        assert state != reduced_state
+
+    def test_expand(
+        self,
+        state: State,
+    ):
+        builder = StateBuilder(
+            state.get_frame(),
+            [
+                CartesianPosition.default(),
+                CartesianVelocity.default(),
+                CoordinatesSubset.mass(),
+            ],
+        )
+        default_state: State = State(
+            state.get_instant(),
+            [100],
+            state.get_frame(),
+            CoordinatesBroker([CoordinatesSubset.mass()]),
+        )
+        expanded_state: State = builder.expand(state, default_state)
+
+        assert isinstance(expanded_state, State)
+        assert state != expanded_state
+        assert default_state != expanded_state
+
+    def test_getters(
+        self,
+        state_builder: StateBuilder,
+        frame: Frame,
+        coordinates_broker: CoordinatesBroker,
+    ):
+        assert state_builder.get_frame() == frame
+        assert state_builder.get_coordinates_subsets() == coordinates_broker.get_subsets()

ostk/astrodynamics/viewer.py

@@ -231,9 +231,7 @@ def _generate_sampled_orientation(states: list[State]) -> cesiumpy.SampledProper
         samples=[
             (
                 coerce_to_datetime(state.get_instant()),
-                _cesium_from_ostk_quaternion(
-                    state.in_frame(Frame.ITRF()).get_attitude()
-                ),
+                _cesium_from_ostk_quaternion(state.in_frame(Frame.ITRF()).get_attitude()),
                 None,
             )
             for state in states

open_space_toolkit_astrodynamics-5.1.5.dist-info/RECORD

@@ -1,25 +0,0 @@
-ostk/astrodynamics/OpenSpaceToolkitAstrodynamicsPy.cpython-38-x86_64-linux-gnu.so,sha256=wQ4lt2wILtqpQcbZ2tT2CvLMNNlFAnIoijcVdI8dNpw,1984248
-ostk/astrodynamics/__init__.py,sha256=3gWyqFIbhAfcdeMhmfBPQPlPQTmaOzm-6flkJe745Zk,251
-ostk/astrodynamics/converters.py,sha256=WIOxUOFGrdFDIKfu0-N8jmCF0GOMMsw2aB_Wy0eXHAQ,3500
-ostk/astrodynamics/display.py,sha256=y9FnoQbPFGM6LzkUdgXgeqtuVGhv57GuTKbeDdcFPgw,6306
-ostk/astrodynamics/libopen-space-toolkit-astrodynamics.so.5,sha256=iNP_YTGSYYFffRaQNGY5-V17l2tqSH-OmLqQnQQ_lLA,93370440
-ostk/astrodynamics/utilities.py,sha256=NbeKN_CtcCRHIinhoNWTEosBMR7xc924S-ZFD97mduw,4478
-ostk/astrodynamics/viewer.py,sha256=AYVm4FIaedpdEvxUb36LSISKYNVnNaQtIDuEqrzLTqI,8948
-ostk/astrodynamics/pytrajectory/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
-ostk/astrodynamics/pytrajectory/pystate.py,sha256=pRWsDnGp23wqXIldaUskYvfbrO3ZsuZSSr4x5Hp-PYY,1067
-ostk/astrodynamics/test/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
-ostk/astrodynamics/test/conftest.py,sha256=J3SbQibnMVaPktAF6Z6cwVCa7xhW3uXpkOJRom6YtCo,2719
-ostk/astrodynamics/test/test_access.py,sha256=s0Lt4MOdcHJasxNuDiDhZJZU-BsY4xBTFYkG4PnXk84,3987
-ostk/astrodynamics/test/test_converters.py,sha256=E5PxWL9AzzCgyBZ2C_0MGpz8sXUW30GJv-LGM7LO5nA,5859
-ostk/astrodynamics/test/test_display.py,sha256=8YeiYRYMpokzqprbRoDBI_xhHI5B8uxcdSzmqmF3AIc,3738
-ostk/astrodynamics/test/test_event_condition.py,sha256=mhMTH7wAoYFWRYt_8l2d1vjNPrFhVjMAEET4INLCVXY,1472
-ostk/astrodynamics/test/test_import.py,sha256=stS8jK9HiXyzRWjtDvaTqUrzCrOVnYVX5CkmQf7MJsA,1201
-ostk/astrodynamics/test/test_root_solver.py,sha256=hQ8O6g-WP49gZH_H3Rdufv0F0gQorpzJyIcjBGGUQ34,1831
-ostk/astrodynamics/test/test_trajectory.py,sha256=pvv4GQsvtU0XfWYk-F-oQNgo8ciFQBgzeJ4FojZDW-A,1364
-ostk/astrodynamics/test/test_utilities.py,sha256=edAPLXNMflCPa0gzhU4e05pa6HbAq7UV0JzMMRx3VlQ,3252
-ostk/astrodynamics/test/test_viewer.py,sha256=x_U6kmxaw_8JtamTbzNj0U-VCFAskWOL87eezv02uAE,3969
-open_space_toolkit_astrodynamics-5.1.5.dist-info/METADATA,sha256=Y5PCQAQ1GlsvU8W12tBkRCz62RRtk5B50xYY4Iq-3Ys,1777
-open_space_toolkit_astrodynamics-5.1.5.dist-info/WHEEL,sha256=LNQH4F1F9_Ua4GgazGQDFGQhQ8B1tSrkPIuEzIRv9nc,93
-open_space_toolkit_astrodynamics-5.1.5.dist-info/top_level.txt,sha256=zOR18699uDYnafgarhL8WU_LmTZY_5NVqutv-flp_x4,5
-open_space_toolkit_astrodynamics-5.1.5.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-open_space_toolkit_astrodynamics-5.1.5.dist-info/RECORD,,