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

ostk/astrodynamics/test/test_converters.py

@@ -1,5 +1,7 @@
 # Apache License 2.0
 
+import pytest
+
 from datetime import datetime, timedelta, timezone
 
 import numpy as np
@@ -17,6 +19,7 @@ from ostk.physics.coordinate import Frame
 
 from ostk.astrodynamics.converters import coerce_to_datetime
 from ostk.astrodynamics.converters import coerce_to_instant
+from ostk.astrodynamics.converters import coerce_to_iso
 from ostk.astrodynamics.converters import coerce_to_interval
 from ostk.astrodynamics.converters import coerce_to_duration
 from ostk.astrodynamics.converters import coerce_to_position
@@ -25,27 +28,231 @@ from ostk.astrodynamics.converters import coerce_to_quaternion
 
 
 def test_coerce_to_datetime_success_instant():
-    value = Instant.date_time(DateTime(2020, 1, 1), Scale.UTC)
-    assert coerce_to_datetime(value) == datetime(2020, 1, 1, tzinfo=timezone.utc)
+    value = Instant.date_time(DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.UTC)
+    assert coerce_to_datetime(value) == datetime(
+        2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone.utc
+    )
+
+    value = Instant.date_time(DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.TAI)
+    assert coerce_to_datetime(value) == datetime(
+        2020, 1, 2, 3, 3, 28, 123456, tzinfo=timezone.utc
+    )
 
 
 def test_coerce_to_datetime_success_datetime():
-    value = datetime(2020, 1, 1, tzinfo=timezone.utc)
+    value = datetime(2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone.utc)
     assert coerce_to_datetime(value) == value
 
+    value = datetime(
+        2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone(timedelta(seconds=3600))
+    )
+    assert coerce_to_datetime(value) == value
+
+
+def test_coerce_to_datetime_success_iso():
+    value = "2020-01-02T03:04:05.123456+00:00"
+    assert coerce_to_datetime(value) == datetime(
+        2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone.utc
+    )
+
+    value = "2020-01-02T03:04:05+00:00"
+    assert coerce_to_datetime(value) == datetime(2020, 1, 2, 3, 4, 5, tzinfo=timezone.utc)
+
+    value = "2020-01-02T03:04:05.123456+01:00"
+    assert coerce_to_datetime(value) == datetime(
+        2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone(timedelta(seconds=3600))
+    )
+
+    value = "2020-01-02T03:04:05.123456Z"
+    assert coerce_to_datetime(value) == datetime(
+        2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone.utc
+    )
+
+    value = "2020-01-02T03:04:05Z"
+    assert coerce_to_datetime(value) == datetime(2020, 1, 2, 3, 4, 5, tzinfo=timezone.utc)
+
+
+def test_coerce_to_datetime_failure():
+    with pytest.raises(TypeError):
+        coerce_to_datetime(False)
+
+    with pytest.raises(Exception):
+        coerce_to_datetime("some_ill_formed_iso")
+
 
 def test_coerce_to_instant_success_datetime():
-    value = datetime(2020, 1, 1, tzinfo=timezone.utc)
+    value = datetime(2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone.utc)
+    assert coerce_to_instant(value) == Instant.date_time(
+        DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.UTC
+    )
+
+    value = datetime(
+        2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone(timedelta(seconds=3600))
+    )
     assert coerce_to_instant(value) == Instant.date_time(
-        DateTime(2020, 1, 1), Scale.UTC
+        DateTime(2020, 1, 2, 2, 4, 5, 123, 456), Scale.UTC
     )
 
 
 def test_coerce_to_instant_success_instant():
-    value = Instant.date_time(DateTime(2020, 1, 1), Scale.UTC)
+    value = Instant.date_time(DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.UTC)
+    assert coerce_to_instant(value) == value
+
+    value = Instant.date_time(DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.TAI)
     assert coerce_to_instant(value) == value
 
 
+def test_coerce_to_instant_success_iso():
+    value = "2020-01-02T03:04:05.123456+00:00"
+    assert coerce_to_instant(value) == Instant.date_time(
+        DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.UTC
+    )
+
+    value = "2020-01-02T03:04:05+00:00"
+    assert coerce_to_instant(value) == Instant.date_time(
+        DateTime(2020, 1, 2, 3, 4, 5), Scale.UTC
+    )
+
+    value = "2020-01-02T03:04:05.123456+01:00"
+    assert coerce_to_instant(value) == Instant.date_time(
+        DateTime(2020, 1, 2, 2, 4, 5, 123, 456), Scale.UTC
+    )
+
+    value = "2020-01-02T03:04:05.123456Z"
+    assert coerce_to_instant(value) == Instant.date_time(
+        DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.UTC
+    )
+
+    value = "2020-01-02T03:04:05Z"
+    assert coerce_to_instant(value) == Instant.date_time(
+        DateTime(
+            2020,
+            1,
+            2,
+            3,
+            4,
+            5,
+        ),
+        Scale.UTC,
+    )
+
+
+def test_coerce_to_instant_failure():
+    with pytest.raises(TypeError):
+        coerce_to_instant(False)
+
+    with pytest.raises(Exception):
+        coerce_to_instant("some_ill_formed_iso")
+
+
+def test_coerce_to_iso_success_datetime():
+    value = datetime(2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone.utc)
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.123456+00:00"
+    )
+
+    value = datetime(
+        2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone(timedelta(seconds=3600))
+    )
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.123456+01:00"
+    )
+
+    value = datetime(2020, 1, 2, 3, 4, 5, 123456, tzinfo=timezone.utc)
+    assert (
+        coerce_to_iso(value, timespec="milliseconds") == "2020-01-02T03:04:05.123+00:00"
+    )
+
+    value = datetime(2020, 1, 2, 3, 4, 5, 0, tzinfo=timezone.utc)
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.000000+00:00"
+    )
+
+
+def test_coerce_to_iso_success_instant():
+    value = Instant.date_time(DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.UTC)
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.123456+00:00"
+    )
+
+    value = Instant.date_time(DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.TAI)
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:03:28.123456+00:00"
+    )
+
+    value = Instant.date_time(DateTime(2020, 1, 2, 3, 4, 5, 123, 456), Scale.UTC)
+    assert (
+        coerce_to_iso(value, timespec="milliseconds") == "2020-01-02T03:04:05.123+00:00"
+    )
+
+    value = Instant.date_time(DateTime(2020, 1, 2, 3, 4, 5, 123), Scale.UTC)
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.123000+00:00"
+    )
+
+
+def test_coerce_to_iso_success_iso():
+    value = "2020-01-02T03:04:05.123456+00:00"
+    assert coerce_to_iso(value, timespec="microseconds") == value
+
+    value = "2020-01-02T03:04:05.123456+01:00"
+    assert coerce_to_iso(value, timespec="microseconds") == value
+
+    value = "2020-01-02T03:04:05.123456+00:00"
+    assert (
+        coerce_to_iso(value, timespec="milliseconds") == "2020-01-02T03:04:05.123+00:00"
+    )
+
+    value = "2020-01-02T03:04:05.123+00:00"
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.123000+00:00"
+    )
+
+    value = "2020-01-02T03:04:05+00:00"
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.000000+00:00"
+    )
+
+    value = "2020-01-02T03:04:05.123456Z"
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.123456+00:00"
+    )
+
+    value = "2020-01-02T03:04:05.123456Z"
+    assert (
+        coerce_to_iso(value, timespec="milliseconds") == "2020-01-02T03:04:05.123+00:00"
+    )
+
+    value = "2020-01-02T03:04:05.123Z"
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.123000+00:00"
+    )
+
+    value = "2020-01-02T03:04:05Z"
+    assert (
+        coerce_to_iso(value, timespec="microseconds")
+        == "2020-01-02T03:04:05.000000+00:00"
+    )
+
+
+def test_coerce_to_iso_failure():
+    with pytest.raises(TypeError):
+        coerce_to_iso(False)
+
+    with pytest.raises(Exception):
+        coerce_to_iso("some_ill_formed_iso")
+
+
 def test_coerce_to_interval_success_interval():
     value = Interval.closed(
         Instant.date_time(DateTime(2020, 1, 1), Scale.UTC),

ostk/astrodynamics/test/test_viewer.py

@@ -103,8 +103,7 @@ class TestViewer:
         assert "Cesium.IonResource.fromAssetId(123)" in rendered_html
         assert "widget.entities.add({polyline:" in rendered_html
         assert (
-            "widget.entities.add({position: widget.star_tracker_position"
-            in rendered_html
+            "widget.entities.add({position: widget.star_tracker_position" in rendered_html
         )
         assert rendered_html.endswith("</script>")
 

ostk/astrodynamics/test/trajectory/__init__.py

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

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

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

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

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

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

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

ostk/astrodynamics/test/trajectory/orbit/messages/spacex/conftest.py

@@ -0,0 +1,18 @@
+# Apache License 2.0
+
+import pytest
+
+import pathlib
+
+from ostk.core.filesystem import Path
+from ostk.core.filesystem import File
+
+
+@pytest.fixture
+def data_directory_path() -> str:
+    return f"{pathlib.Path(__file__).parent.absolute()}/data"
+
+
+@pytest.fixture
+def opm_file(data_directory_path: str) -> File:
+    return File.path(Path.parse(f"{data_directory_path}/opm_1.yaml"))

ostk/astrodynamics/test/trajectory/orbit/messages/spacex/data/opm_1.yaml

@@ -0,0 +1,44 @@
+# Dummy SpaceX OPM output
+
+# Notes:
+# - ECEF velocity is Earth relative
+# - Apogee/Perigee altitude assumes a spherical Earth, 6378.137 km radius
+# - Orbital elements are computed in an inertial frame realized by inertially
+#   freezing the WGS84 ECEF frame at time of current state
+# - State is post-deployment, so includes separation delta-velocity
+
+
+header:
+  generation_date: 2020-01-01T12:34:56.789Z
+  launch_date: 2020-01-02T12:34:56.789Z
+
+
+deployments:
+
+- name: satellite_a
+  sequence_number: 1
+  mission_time_s: 3600.0
+  date: 2020-01-02T13:34:56.789Z
+  r_ecef_m: [693289.644, 6876578.628, -133035.288]
+  v_ecef_m_per_s: [1305.783, 39.783, 7525.920]
+  mean_perigee_altitude_km: 526.768
+  mean_apogee_altitude_km: 568.430
+  mean_inclination_deg: 97.123
+  mean_argument_of_perigee_deg: -179.513
+  mean_longitude_ascending_node_deg: 85.057
+  mean_mean_anomaly_deg: 179.263
+  ballistic_coef_kg_per_m2: 47.55
+
+- name: satellite_b
+  sequence_number: 2
+  mission_time_s: 7200.0
+  date: 2020-01-02T14:34:56.789Z
+  r_ecef_m: [699863.059, 6875647.517, -123777.595]
+  v_ecef_m_per_s: [1504.658, 6.705, 7538.669]
+  mean_perigee_altitude_km: 536.779
+  mean_apogee_altitude_km: 529.851
+  mean_inclination_deg: 97.124
+  mean_argument_of_perigee_deg: 136.875
+  mean_longitude_ascending_node_deg: 85.032
+  mean_mean_anomaly_deg: -127.164
+  ballistic_coef_kg_per_m2: 44.26

ostk/astrodynamics/test/trajectory/orbit/messages/spacex/test_opm.py

@@ -0,0 +1,108 @@
+# Apache License 2.0
+
+import pytest
+
+from datetime import datetime
+
+from ostk.core.containers import Dictionary
+from ostk.core.filesystem import Path
+from ostk.core.filesystem import File
+
+from ostk.physics.units import Length
+from ostk.physics.units import Angle
+from ostk.physics.time import Instant
+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.astrodynamics.trajectory.orbit.messages.spacex import OPM
+
+
+@pytest.fixture
+def opm() -> OPM:
+    return OPM(
+        header=OPM.Header(
+            generation_date=Instant.date_time(datetime(2020, 1, 2, 3, 4, 5), Scale.UTC),
+            launch_date=Instant.date_time(datetime(2020, 1, 3, 3, 4, 5), Scale.UTC),
+        ),
+        deployments=[
+            OPM.Deployment(
+                name="A",
+                sequence_number=1,
+                mission_time=Duration.hours(3600.0),
+                date=Instant.date_time(datetime(2020, 1, 4, 3, 4, 5), Scale.UTC),
+                position=Position.meters((1.0, 2.0, 3.0), Frame.ITRF()),
+                velocity=Velocity.meters_per_second((4.0, 5.0, 6.0), Frame.ITRF()),
+                mean_perigee_altitude=Length.kilometers(500.0),
+                mean_apogee_altitude=Length.kilometers(500.0),
+                mean_inclination=Angle.degrees(1.0),
+                mean_argument_of_perigee=Angle.degrees(2.0),
+                mean_longitude_ascending_node=Angle.degrees(3.0),
+                mean_mean_anomaly=Angle.degrees(4.0),
+                ballistic_coefficient=123.456,
+            ),
+        ],
+    )
+
+
+class TestOPM:
+    def test_constructor(self, opm: OPM):
+        assert opm is not None
+
+    def test_get_header(self, opm: OPM):
+        assert opm.get_header() is not None
+
+    def test_get_deployments(self, opm: OPM):
+        assert len(opm.get_deployments()) == 1
+
+    def test_get_deployment_at(self, opm: OPM):
+        assert opm.get_deployment_at(index=0).name == "A"
+
+    def test_get_deployment_with_name(self, opm: OPM):
+        assert opm.get_deployment_with_name(name="A").name == "A"
+
+    def test_undefined(self):
+        assert OPM.undefined().is_defined() is False
+
+    def test_dictionary(self):
+        dictionary = Dictionary(
+            {
+                "header": {
+                    "generation_date": "2020-01-01T12:34:56.789Z",
+                    "launch_date": "2020-01-02T12:34:56.789Z",
+                },
+                "deployments": [
+                    {
+                        "name": "satellite_a",
+                        "sequence_number": 1,
+                        "mission_time_s": 3600.0,
+                        "date": "2020-01-02T13:34:56.789Z",
+                        "r_ecef_m": [693289.644, 6876578.628, -133035.288],
+                        "v_ecef_m_per_s": [1305.783, 39.783, 7525.920],
+                        "mean_perigee_altitude_km": 526.768,
+                        "mean_apogee_altitude_km": 568.430,
+                        "mean_inclination_deg": 97.123,
+                        "mean_argument_of_perigee_deg": -179.513,
+                        "mean_longitude_ascending_node_deg": 85.057,
+                        "mean_mean_anomaly_deg": 179.263,
+                        "ballistic_coef_kg_per_m2": 47.55,
+                    }
+                ],
+            }
+        )
+
+        assert OPM.dictionary(dictionary) is not None
+
+    def test_parse(self, opm_file: File):
+        with open(str(opm_file.get_path().to_string()), "r") as stream:
+            assert OPM.parse(string=stream.read()) is not None
+
+    def test_load(self, opm_file: File):
+        assert OPM.load(file=opm_file) is not None
+
+
+class TestOPMDeployment:
+    def test_to_state(self, opm: OPM):
+        assert opm.get_deployment_with_name(name="A").to_state() is not None

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

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

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

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

ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_brouwer_lyddane_mean.py

@@ -0,0 +1,65 @@
+# Apache License 2.0
+
+import pytest
+
+from ostk.physics.units import Length
+from ostk.physics.units import Angle
+
+from ostk.astrodynamics.trajectory.orbit.models import BrouwerLyddaneMean
+
+
+@pytest.fixture
+def semi_major_axis() -> Length:
+    return Length.kilometers(7000.0)
+
+
+@pytest.fixture
+def eccentricity() -> float:
+    return 1e-3
+
+
+@pytest.fixture
+def inclination() -> Angle:
+    return Angle.degrees(35.0)
+
+
+@pytest.fixture
+def raan() -> Angle:
+    return Angle.degrees(40.0)
+
+
+@pytest.fixture
+def aop() -> Angle:
+    return Angle.degrees(50.0)
+
+
+@pytest.fixture
+def mean_anomaly() -> Angle:
+    return Angle.degrees(60.0)
+
+
+@pytest.fixture
+def brouwer_lyddane_mean(
+    semi_major_axis, eccentricity, inclination, raan, aop, mean_anomaly
+):
+    class BrouwerLyddaneMeanMock(BrouwerLyddaneMean):
+        def __init__(
+            self, semi_major_axis, eccentricity, inclination, raan, aop, mean_anomaly
+        ):
+            super().__init__(
+                semi_major_axis, eccentricity, inclination, raan, aop, mean_anomaly
+            )
+
+        def to_coe():
+            return 0.0
+
+    return BrouwerLyddaneMeanMock(
+        semi_major_axis, eccentricity, inclination, raan, aop, mean_anomaly
+    )
+
+
+class TestBrouwerLyddaneMean:
+    def test_getters(self, brouwer_lyddane_mean: BrouwerLyddaneMean):
+        assert isinstance(brouwer_lyddane_mean.get_mean_anomaly(), Angle)
+        assert isinstance(brouwer_lyddane_mean.get_true_anomaly(), Angle)
+        assert isinstance(brouwer_lyddane_mean.get_eccentric_anomaly(), Angle)

ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_brouwer_lyddane_mean_long.py

@@ -0,0 +1,102 @@
+# Apache License 2.0
+
+import pytest
+
+from ostk.physics.units import Length
+from ostk.physics.units import Angle
+from ostk.physics.environment.gravitational import Earth
+
+from ostk.physics.coordinate import Frame
+from ostk.physics.coordinate import Position
+from ostk.physics.coordinate import Velocity
+
+from ostk.astrodynamics.trajectory.orbit.models.kepler import COE
+from ostk.astrodynamics.trajectory.orbit.models.brouwerLyddaneMean import (
+    BrouwerLyddaneMeanLong,
+)
+
+
+@pytest.fixture
+def semi_major_axis() -> Length:
+    return Length.kilometers(7000.0)
+
+
+@pytest.fixture
+def eccentricity() -> float:
+    return 1e-3
+
+
+@pytest.fixture
+def inclination() -> Angle:
+    return Angle.degrees(35.0)
+
+
+@pytest.fixture
+def raan() -> Angle:
+    return Angle.degrees(40.0)
+
+
+@pytest.fixture
+def aop() -> Angle:
+    return Angle.degrees(50.0)
+
+
+@pytest.fixture
+def mean_anomaly() -> Angle:
+    return Angle.degrees(60.0)
+
+
+@pytest.fixture
+def coe(
+    semi_major_axis: Length,
+    eccentricity: float,
+    inclination: Angle,
+    raan: Angle,
+    aop: Angle,
+    mean_anomaly: Angle,
+) -> COE:
+    return COE(semi_major_axis, eccentricity, inclination, raan, aop, mean_anomaly)
+
+
+@pytest.fixture
+def cartesian_state() -> tuple[Position, Velocity]:
+    return (
+        Position.meters(
+            [6596407.223662058, 2281266.582975321, -10540.61521486086], Frame.GCRF()
+        ),
+        Velocity.meters_per_second(
+            [337.7269674273224, -969.7192552349448, 7488.702816619139], Frame.GCRF()
+        ),
+    )
+
+
+@pytest.fixture
+def gravitational_parameter():
+    return Earth.EGM2008.gravitational_parameter
+
+
+@pytest.fixture
+def brouwer_lyddane_mean_long(
+    semi_major_axis, eccentricity, inclination, raan, aop, mean_anomaly
+):
+    return BrouwerLyddaneMeanLong(
+        semi_major_axis, eccentricity, inclination, raan, aop, mean_anomaly
+    )
+
+
+class TestBrouwerLyddaneMeanLong:
+    def test_to_coe(self, brouwer_lyddane_mean_long: BrouwerLyddaneMeanLong):
+        assert brouwer_lyddane_mean_long.to_coe().is_defined()
+
+    def test_coe(self, coe: COE):
+        assert BrouwerLyddaneMeanLong.COE(coe).is_defined()
+
+    def test_cartesian(
+        self, cartesian_state: tuple[Position, Velocity], gravitational_parameter
+    ):
+        assert BrouwerLyddaneMeanLong.cartesian(
+            cartesian_state, gravitational_parameter
+        ).is_defined()
+
+    def test_undefined(self):
+        assert BrouwerLyddaneMeanLong.undefined().is_defined() is False