open-space-toolkit-physics 11.3.0__py313-none-manylinux2014_x86_64.whl

ostk/physics/test/coordinate/test_velocity.py

@@ -0,0 +1,180 @@
+# Apache License 2.0
+
+import pytest
+import numpy as np
+
+from ostk.core.type import String
+import ostk.physics as physics
+
+Scale = physics.time.Scale
+DateTime = physics.time.DateTime
+Instant = physics.time.Instant
+Frame = physics.coordinate.Frame
+Velocity = physics.coordinate.Velocity
+Unit = Velocity.Unit
+
+
+def test_coordinate_position_constructors():
+    # Construct arbitrary frame
+    frame: Frame = Frame.GCRF()
+    unit: Unit = Unit.MeterPerSecond
+
+    # Construction with python lists
+    vector = [1000.0, 0.0, 0.0]
+
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    assert velocity is not None
+    assert isinstance(velocity, Velocity)
+    assert velocity.is_defined()
+
+    # Construction with python tuples
+    vector = (1000.0, 0.0, 0.0)
+
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    assert velocity is not None
+    assert isinstance(velocity, Velocity)
+    assert velocity.is_defined()
+
+    # Construction with python numpy arrays
+    vector = np.array((1000.0, 0.0, 0.0))
+
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    assert velocity is not None
+    assert isinstance(velocity, Velocity)
+    assert velocity.is_defined()
+
+    # Construction with meters static constructor
+    velocity: Velocity = Velocity.meters_per_second(vector, frame)
+
+    assert velocity is not None
+    assert isinstance(velocity, Velocity)
+    assert velocity.is_defined()
+
+    # Construction with undefined static constructor
+    velocity: Velocity = Velocity.undefined()
+
+    assert velocity is not None
+    assert isinstance(velocity, Velocity)
+    assert velocity.is_defined() is False
+
+
+def test_coordinate_velocity_comparators():
+    frame: Frame = Frame.GCRF()
+    unit: Unit = Unit.MeterPerSecond
+    vector_1 = [1000.0, 0.0, 0.0]
+    vector_2 = [-450.3, 234.9, -23.0]
+
+    velocity_1: Velocity = Velocity(vector_1, unit, frame)
+    velocity_2: Velocity = Velocity(vector_2, unit, frame)
+
+    assert velocity_1 == velocity_1
+    assert velocity_2 == velocity_2
+    assert velocity_1 != velocity_2
+
+
+def test_coordinate_velocity_is_defined():
+    frame: Frame = Frame.GCRF()
+    unit: Unit = Unit.MeterPerSecond
+    vector = [1000.0, 0.0, 0.0]
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    assert velocity.is_defined()
+    assert Velocity.undefined().is_defined() is False
+
+
+def test_coordinate_velocity_access_frame():
+    frame: Frame = Frame.GCRF()
+    unit: Unit = Unit.MeterPerSecond
+    vector = [1000.0, 0.0, 0.0]
+
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    ans_frame = velocity.access_frame()
+
+    assert ans_frame is not None
+    assert isinstance(ans_frame, Frame)
+    assert ans_frame == frame
+
+
+def test_coordinate_velocity_get_coordinates():
+    frame: Frame = Frame.GCRF()
+    unit: Unit = Unit.MeterPerSecond
+    vector = [1000.0, 0.0, 0.0]
+
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    coordinates = velocity.get_coordinates()
+
+    assert coordinates is not None
+    assert isinstance(coordinates, np.ndarray)
+    assert np.array_equal(coordinates, vector)
+
+
+def test_coordinate_velocity_get_unit():
+    frame: Frame = Frame.GCRF()
+    unit: Unit = Unit.MeterPerSecond
+    vector = [1000.0, 0.0, 0.0]
+
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    ans_unit = velocity.get_unit()
+
+    assert ans_unit is not None
+    assert isinstance(ans_unit, Unit)
+    assert ans_unit == unit
+
+
+def test_coordinate_velocity_in_unit():
+    frame: Frame = Frame.GCRF()
+    unit: Unit = Unit.MeterPerSecond
+    vector = [1000.0, 0.0, 0.0]
+
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    new_velocity: Velocity = velocity.in_unit(Unit.MeterPerSecond)
+
+    assert new_velocity is not None
+    assert isinstance(new_velocity, Velocity)
+    assert new_velocity == velocity
+
+
+@pytest.mark.skip
+def test_coordinate_velocity_in_frame():
+    frame: Frame = Frame.GCRF()
+    unit: Unit = Unit.MeterPerSecond
+    vector = [1000.0, 0.0, 0.0]
+
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    instant: Instant = Instant.date_time(DateTime(2020, 1, 1, 0, 0, 0), Scale.UTC)
+
+    new_velocity: Velocity = velocity.in_frame(frame, instant)
+
+    assert new_velocity is not None
+    assert isinstance(new_velocity, Velocity)
+    assert new_velocity == velocity
+
+
+def test_coordinate_velocity_to_string():
+    frame: Frame = Frame.GCRF()
+    unit: Unit = Unit.MeterPerSecond
+    vector = [1000.0, 0.0, 0.0]
+
+    velocity: Velocity = Velocity(vector, unit, frame)
+
+    string: String = velocity.to_string()
+
+    assert string is not None
+    assert isinstance(string, String)
+    assert string == "[1000.0, 0.0, 0.0] [m/s] @ GCRF"
+
+
+def test_coordinate_velocity_string_from_unit():
+    string: String = Velocity.string_from_unit(Unit.MeterPerSecond)
+
+    assert string is not None
+    assert isinstance(string, String)
+    assert string == "m/s"

ostk/physics/test/data/conftest.py

@@ -0,0 +1,34 @@
+# Apache License 2.0
+
+import pytest
+
+import pathlib
+
+from ostk.core.filesystem import Path
+from ostk.core.filesystem import File
+
+from ostk.physics.data import Manager, Manifest
+
+
+@pytest.fixture
+def data_directory_path() -> str:
+    return f"{pathlib.Path(__file__).parent.absolute()}/data"
+
+
+@pytest.fixture
+def manifest_file(data_directory_path: str) -> File:
+    return File.path(Path.parse(f"{data_directory_path}/manifest.json"))
+
+
+@pytest.fixture
+def manifest(manifest_file: File) -> Manifest:
+    return Manifest.load(manifest_file)
+
+
+@pytest.fixture
+def manager() -> Manager:
+    manager = Manager.get()
+
+    yield manager
+
+    manager.reset()

ostk/physics/test/data/data/manifest.json

@@ -0,0 +1,22 @@
+{
+    "manifest": {
+        "path": "",
+        "filenames": "manifest.json",
+        "file_meta": "This file. Entry tracks the expected manifest update time on the github so that local OSTk can make smarter pulling decisions.",
+        "last_update": "2023-11-01T22:05:43.789048",
+        "next_update_check": "2023-11-02T04:05:43.789093",
+        "check_frequency": "6 hours"
+    },
+    "bulletin-A": {
+        "path": "coordinate/frame/providers/iers/bulletin-A",
+        "filenames": "ser7.dat",
+        "remote_sources": [
+            {
+                "url": "https://maia.usno.navy.mil/ser7/ser7.dat"
+            }
+        ],
+        "last_update": "2023-08-04T12:02:17.028701",
+        "next_update_check": "2023-08-05T12:02:17.029315",
+        "check_frequency": "1 day"
+    }
+}

ostk/physics/test/data/provider/test_provider.py

@@ -0,0 +1,34 @@
+# Apache License 2.0
+
+import pytest
+
+from ostk.physics import Environment
+from ostk.physics.environment.object import Celestial
+from ostk.physics.coordinate import Position
+from ostk.physics.coordinate import Frame
+from ostk.physics.data import provider
+
+
+@pytest.fixture
+def environment() -> Environment:
+    return Environment.default()
+
+
+@pytest.fixture
+def celestial(environment: Environment) -> Celestial:
+    return environment.access_celestial_object_with_name("Earth")
+
+
+@pytest.fixture
+def position():
+    return Position.meters([6671000.0, 0.0, 0.0], Frame.ITRF())
+
+
+class TestProvider:
+    def test_nadir(
+        self,
+        celestial: Celestial,
+        position: Position,
+        environment: Environment,
+    ):
+        assert provider.nadir(position, celestial, environment) is not None

ostk/physics/test/data/test_direction.py

@@ -0,0 +1,40 @@
+# Apache License 2.0
+
+import pytest
+
+import numpy as np
+
+from ostk.physics.data import Direction
+from ostk.physics.coordinate import Frame
+
+
+@pytest.fixture
+def value() -> np.ndarray:
+    return np.array([1.0, 0.0, 0.0])
+
+
+@pytest.fixture
+def frame() -> Frame:
+    return Frame.GCRF()
+
+
+@pytest.fixture
+def direction(value: float, frame: Frame) -> Direction:
+    return Direction(
+        value=value,
+        frame=frame,
+    )
+
+
+class TestDirection:
+    def test_constructor(self, direction: Direction):
+        assert direction is not None
+
+    def test_equals(self, direction: Direction):
+        assert direction == direction
+
+    def test_not_equals(self, direction: Direction):
+        assert (direction != direction) is False
+
+    def test_undefined(self):
+        assert Direction.undefined() is not None

ostk/physics/test/data/test_manifest.py

@@ -0,0 +1,43 @@
+# Apache License 2.0
+
+import pytest
+from datetime import datetime
+
+from ostk.core.filesystem import File
+from ostk.core.filesystem import Path
+
+from ostk.io import URL
+
+from ostk.physics.data import Manifest
+from ostk.physics.time import Instant
+
+
+class TestManifest:
+    def test_is_defined_success(self, manifest: Manifest):
+        assert manifest.is_defined() == True
+
+    def test_get_last_modified_timestamp_success(self, manifest: Manifest):
+        assert isinstance(manifest.get_last_modified_timestamp(), Instant)
+
+    def test_get_last_update_timestamp_for_success(self, manifest: Manifest):
+        assert isinstance(manifest.get_last_update_timestamp_for("manifest"), Instant)
+
+    def test_get_next_update_check_timestamp_for_success(self, manifest: Manifest):
+        assert isinstance(
+            manifest.get_next_update_check_timestamp_for("manifest"), Instant
+        )
+
+    def test_get_remote_data_urls_success(self, manifest: Manifest):
+        test_url: URL = URL.parse("http://test.com")
+        assert isinstance(manifest.get_remote_data_urls(test_url, "manifest"), list)
+
+    def test_find_remote_data_urls_success(self, manifest: Manifest):
+        test_url: URL = URL.parse("http://test.com")
+        assert isinstance(manifest.find_remote_data_urls(test_url, ".*"), list)
+
+    def test_undefined_success(self):
+        assert isinstance(Manifest.undefined(), Manifest)
+
+    def test_load_success(self, manifest_file: File):
+        manifest = Manifest.load(manifest_file)
+        assert isinstance(manifest, Manifest)

ostk/physics/test/data/test_manifest_manager.py

@@ -0,0 +1,62 @@
+# Apache License 2.0
+
+import pytest
+from datetime import datetime
+from ostk.core.filesystem import Directory
+from ostk.core.filesystem import File
+from ostk.core.filesystem import Path
+
+from ostk.io import URL
+
+from ostk.physics.time import Instant
+from ostk.physics.data import Manager
+from ostk.physics.data import Manifest
+
+
+class TestManager:
+    def test_get_last_update_timestamp_for_success(self, manager: Manager):
+        assert isinstance(manager.get_last_update_timestamp_for("manifest"), Instant)
+
+    def test_get_remote_url_success(self, manager: Manager):
+        assert isinstance(manager.get_remote_url(), URL)
+
+    def test_get_local_repository_success(self, manager: Manager):
+        assert isinstance(manager.get_local_repository(), Directory)
+
+    def test_get_remote_data_urls_success(self, manager: Manager):
+        assert isinstance(manager.get_remote_data_urls("data_name"), list)
+
+    def test_get_manifest_success(self, manager: Manager):
+        assert isinstance(manager.get_manifest(), Manifest)
+
+    def test_set_remote_url_success(self, manager: Manager):
+        manager.set_remote_url(URL.parse("http://example.com"))
+        assert manager.get_remote_url() == URL.parse("http://example.com")
+
+    def test_set_local_repository_success(self, manager: Manager):
+        directory = Directory.path(Path.parse("./test_directory"))
+        manager.set_local_repository(directory)
+        assert manager.get_local_repository() == directory
+
+    def test_find_remote_data_urls_success(self, manager: Manager):
+        assert isinstance(manager.find_remote_data_urls(".*"), list)
+
+    def test_load_manifest_success(self, manager: Manager, manifest: Manifest):
+        manager.load_manifest(manifest)
+        assert manager.get_manifest().is_defined()
+
+    def test_manifest_file_exists_success(self, manager: Manager):
+        assert isinstance(manager.manifest_file_exists(), bool)
+
+    def test_reset_success(self, manager: Manager, manifest: Manifest):
+        manager.load_manifest(manifest)
+        assert manager.get_manifest().is_defined()
+
+        manager.reset()
+        assert not manager.get_manifest().is_defined()
+
+    def test_get_success(self):
+        assert isinstance(Manager.get(), Manager)
+
+    def test_default_remote_url_success(self, manager: Manager):
+        assert isinstance(manager.default_remote_url(), URL)

ostk/physics/test/data/test_scalar.py

@@ -0,0 +1,54 @@
+# Apache License 2.0
+
+import pytest
+
+from ostk.physics import Unit
+from ostk.physics.data import Scalar
+from ostk.physics.unit import Length
+
+
+@pytest.fixture
+def unit() -> Unit:
+    return Unit.length(Length.Unit.Meter)
+
+
+@pytest.fixture
+def value() -> float:
+    return 5.0
+
+
+@pytest.fixture
+def scalar(value: float, unit: Unit) -> Scalar:
+    return Scalar(
+        value=value,
+        unit=unit,
+    )
+
+
+class TestScalar:
+    def test_constructor(self, scalar: Scalar):
+        assert scalar is not None
+
+    def test_equals(self, scalar: Scalar):
+        assert scalar == scalar
+
+    def test_not_equals(self, scalar: Scalar):
+        assert (scalar != scalar) is False
+
+    def test_is_defined(self, scalar: Scalar):
+        assert scalar.is_defined()
+
+    def test_get_value(self, scalar: Scalar, value: float):
+        assert scalar.get_value() == value
+
+    def test_get_unit(self, scalar: Scalar):
+        assert scalar.get_unit().is_defined()
+
+    def test_in_unit(self, scalar: Scalar):
+        assert scalar.in_unit(Unit.length(Length.Unit.Foot)) is not None
+
+    def test_to_string(self, scalar: Scalar):
+        assert scalar.to_string() is not None
+
+    def test_undefined(self):
+        assert Scalar.undefined() is not None

ostk/physics/test/data/test_vector.py

@@ -0,0 +1,70 @@
+# Apache License 2.0
+
+import pytest
+
+import numpy as np
+
+from ostk.physics import Unit
+from ostk.physics.data import Vector
+from ostk.physics.unit import Length
+from ostk.physics.time import Instant
+from ostk.physics.coordinate import Frame
+
+
+@pytest.fixture
+def unit() -> Unit:
+    return Unit.length(Length.Unit.Meter)
+
+
+@pytest.fixture
+def value() -> np.ndarray:
+    return np.array([1.0, 2.0, 3.0])
+
+
+@pytest.fixture
+def frame() -> Frame:
+    return Frame.GCRF()
+
+
+@pytest.fixture
+def vector(value: float, unit: Unit, frame: Frame) -> Vector:
+    return Vector(
+        value=value,
+        unit=unit,
+        frame=frame,
+    )
+
+
+class TestVector:
+    def test_constructor(self, vector: Vector):
+        assert vector is not None
+
+    def test_equals(self, vector: Vector):
+        assert vector == vector
+
+    def test_not_equals(self, vector: Vector):
+        assert (vector != vector) is False
+
+    def test_is_defined(self, vector: Vector):
+        assert vector.is_defined()
+
+    def test_get_value(self, vector: Vector, value: float):
+        assert np.all(vector.get_value() == value)
+
+    def test_get_unit(self, vector: Vector):
+        assert vector.get_unit().is_defined()
+
+    def test_get_frame(self, vector: Vector, frame: Frame):
+        assert vector.get_frame() == frame
+
+    def test_in_unit(self, vector: Vector):
+        assert vector.in_unit(Unit.length(Length.Unit.Foot)) is not None
+
+    def test_in_frame(self, vector: Vector):
+        assert vector.in_frame(Frame.ITRF(), Instant.J2000()) is not None
+
+    def test_to_string(self, vector: Vector):
+        assert vector.to_string() is not None
+
+    def test_undefined(self):
+        assert Vector.undefined() is not None

ostk/physics/test/environment/__init__.py

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

ostk/physics/test/environment/atmospheric/__init__.py

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

ostk/physics/test/environment/atmospheric/earth/__init__.py

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

ostk/physics/test/environment/atmospheric/earth/conftest.py

@@ -0,0 +1,67 @@
+# Apache License 2.0
+
+import pytest
+
+import pathlib
+
+from ostk.core.filesystem import Path
+from ostk.core.filesystem import File
+
+from ostk.io import URL
+
+from ostk.physics.coordinate import Frame
+from ostk.physics.environment.object.celestial import Sun
+from ostk.physics.environment.atmospheric.earth import Manager
+from ostk.physics.environment.atmospheric.earth import CSSISpaceWeather
+from ostk.physics.environment.atmospheric.earth import Exponential
+from ostk.physics.environment.atmospheric.earth import NRLMSISE00
+from ostk.physics.environment.gravitational import Earth as EarthGravityModel
+
+
+@pytest.fixture
+def data_directory_path() -> str:
+    return f"{pathlib.Path(__file__).parent.absolute()}/data"
+
+
+@pytest.fixture
+def cssi_space_weather_file(data_directory_path: str) -> File:
+    return File.path(Path.parse(f"{data_directory_path}/SW-Last5Years.test.csv"))
+
+
+@pytest.fixture
+def cssi_space_weather_legacy_file(data_directory_path: str) -> File:
+    return File.path(Path.parse(f"{data_directory_path}/SpaceWeather-All-v1.2.test.txt"))
+
+
+@pytest.fixture
+def cssi_space_weather(cssi_space_weather_file: File) -> CSSISpaceWeather:
+    return CSSISpaceWeather.load(cssi_space_weather_file)
+
+
+@pytest.fixture
+def manager() -> Manager:
+    manager = Manager.get()
+
+    yield manager
+
+    manager.reset()
+    manager.clear_local_repository()
+
+
+@pytest.fixture
+def exponential_model() -> Exponential:
+    return Exponential()
+
+
+@pytest.fixture
+def nrlmsise00_model() -> NRLMSISE00:
+    return NRLMSISE00(
+        NRLMSISE00.InputDataType.CSSISpaceWeatherFile,
+        160.0,
+        160.0,
+        3.0,
+        Frame.ITRF(),
+        EarthGravityModel.WGS84.equatorial_radius,
+        EarthGravityModel.WGS84.flattening,
+        Sun.spherical(),
+    )

ostk/physics/test/environment/atmospheric/earth/data/SW-Last5Years.test.csv

@@ -0,0 +1,22 @@
+DATE,BSRN,ND,KP1,KP2,KP3,KP4,KP5,KP6,KP7,KP8,KP_SUM,AP1,AP2,AP3,AP4,AP5,AP6,AP7,AP8,AP_AVG,CP,C9,ISN,F10.7_OBS,F10.7_ADJ,F10.7_DATA_TYPE,F10.7_OBS_CENTER81,F10.7_OBS_LAST81,F10.7_ADJ_CENTER81,F10.7_ADJ_LAST81
+2018-01-01,2515,21,33,37,23,23,27,10,10,13,177,18,22,9,9,12,4,4,5,10,0.6,3,0,69.1,66.8,OBS,71.4,72.5,69.3,70.8
+2018-01-02,2515,22,17,7,3,7,17,10,20,10,90,6,3,2,3,6,4,7,4,4,0.2,1,0,69.5,67.2,OBS,71.5,72.5,69.3,70.8
+2018-01-03,2515,23,0,7,10,7,3,10,3,0,40,0,3,4,3,2,4,2,0,2,0.0,0,0,70.7,68.3,INT,71.6,72.5,69.4,70.8
+2023-06-18,2589,17,27,20,13,20,17,17,27,20,160,12,7,5,7,6,6,12,7,8,0.4,2,159,164.1,169.4,OBS,160.4,152.8,165.2,155.5
+2023-06-19,2589,18,23,17,17,13,20,27,30,30,177,9,6,6,5,7,12,15,15,9,0.5,2,179,168.8,174.3,OBS,160.1,153.1,164.9,155.9
+
+2023-06-20,2589,19,27,20,10,17,20,23,10,40,167,12,7,4,6,7,9,4,27,9,0.5,2,203,164.6,170.0,PRD,160.2,153.6,165.0,156.4
+2023-06-21,2589,20,30,37,27,27,20,20,27,27,213,15,22,12,12,7,7,12,12,12,0.7,3,119,164.6,170.0,PRD,160.3,154.0,165.1,157.0
+2023-08-02,2591,8,13,13,13,13,13,13,13,13,104,5,5,5,5,5,5,5,5,5,0.2,1,123,165.0,170.0,PRD,157.5,160.8,162.0,165.7
+2023-08-03,2591,9,13,13,13,13,13,13,13,13,104,5,5,5,5,5,5,5,5,5,0.2,1,123,165.1,170.0,PRD,157.4,161.1,161.8,166.0
+
+2023-09-01,2592,11,,,,,,,,,,,,,,,,,,,,,124,151.3,154.2,PRM,154.2,158.3,156.9,163.1
+2023-10-01,2593,14,,,,,,,,,,,,,,,,,,,,,126,154.0,154.4,PRM,153.9,154.5,154.3,157.9
+2023-11-01,2594,18,,,,,,,,,,,,,,,,,,,,,128,156.6,154.3,PRM,156.4,153.1,154.3,154.2
+2023-12-01,2595,21,,,,,,,,,,,,,,,,,,,,,129,158.5,154.2,PRM,158.2,155.6,154.2,154.3
+2028-08-01,2658,25,,,,,,,,,,,,,,,,,,,,,45,82.7,85.2,PRM,83.3,84.3,85.6,86.9
+2028-09-01,2660,2,,,,,,,,,,,,,,,,,,,,,43,82.7,84.2,PRM,83.2,83.5,84.6,85.9
+2028-10-01,2661,5,,,,,,,,,,,,,,,,,,,,,41,82.9,83.1,PRM,83.5,83.2,83.6,85.0
+2028-11-01,2662,9,,,,,,,,,,,,,,,,,,,,,40,83.3,82.1,PRM,83.8,83.4,82.6,84.0
+2028-12-01,2663,12,,,,,,,,,,,,,,,,,,,,,38,83.7,81.4,PRM,83.9,83.7,81.8,83.0
+2029-01-01,2664,16,,,,,,,,,,,,,,,,,,,,,36,83.5,80.7,PRM,83.6,83.9,81.0,82.0