open-space-toolkit-physics 6.0.4__py311-none-manylinux2014_aarch64.whl → 11.3.0__py311-none-manylinux2014_aarch64.whl

ostk/physics/test/coordinate/test_position.py

@@ -4,203 +4,198 @@ 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
-Position = physics.coordinate.Position
-Unit = physics.unit.Length.Unit
+from ostk.physics import Environment
+from ostk.physics.coordinate import Frame
+from ostk.physics.coordinate import Position
+from ostk.physics.coordinate.spherical import LLA
+from ostk.physics.time import Instant
+from ostk.physics.time import Scale
+from ostk.physics.time import DateTime
+from ostk.physics.unit import Length
 
 
-def test_coordinate_position_constructors():
-    # Construct arbitrary frame
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
+@pytest.fixture
+def unit() -> Length.Unit:
+    return Length.Unit.Meter
 
-    # Construction with python lists
-    vector = [1000.0, 0.0, 0.0]
 
-    position: Position = Position(vector, unit, frame)
+@pytest.fixture
+def frame() -> Frame:
+    return Frame.GCRF()
 
-    assert position is not None
-    assert isinstance(position, Position)
-    assert position.is_defined()
 
-    # Construction with python tuples
-    vector = (1000.0, 0.0, 0.0)
+@pytest.fixture
+def earth():
+    return Environment.default().access_celestial_object_with_name("Earth")
 
-    position: Position = Position(vector, unit, frame)
 
-    assert position is not None
-    assert isinstance(position, Position)
-    assert position.is_defined()
+@pytest.fixture
+def lla() -> LLA:
+    return LLA.vector([0.0, 0.0, 0.0])
 
-    # Construction with python numpy arrays
-    vector = np.array((1000.0, 0.0, 0.0))
 
-    position: Position = Position(vector, unit, frame)
+class TestPosition:
 
-    assert position is not None
-    assert isinstance(position, Position)
-    assert position.is_defined()
+    def test_constructors(self, unit: Length.Unit, frame: Frame):
+        # Construction with python lists
+        vector = [1000.0, 0.0, 0.0]
 
-    # Construction with meters static constructor
-    position: Position = Position.meters(vector, frame)
+        position: Position = Position(vector, unit, frame)
 
-    assert position is not None
-    assert isinstance(position, Position)
-    assert position.is_defined()
+        assert position is not None
+        assert isinstance(position, Position)
+        assert position.is_defined()
 
-    # Construction with undefined static constructor
-    position: Position = Position.undefined()
+        # Construction with python tuples
+        vector = (1000.0, 0.0, 0.0)
 
-    assert position is not None
-    assert isinstance(position, Position)
-    assert position.is_defined() is False
+        position: Position = Position(vector, unit, frame)
 
+        assert position is not None
+        assert isinstance(position, Position)
+        assert position.is_defined()
 
-def test_coordinate_position_comparators():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
-    vector_1 = [1000.0, 0.0, 0.0]
-    vector_2 = [-450.3, 234.9, -23.0]
+        # Construction with python numpy arrays
+        vector = np.array((1000.0, 0.0, 0.0))
 
-    position_1: Position = Position(vector_1, unit, frame)
-    position_2: Position = Position(vector_2, unit, frame)
+        position: Position = Position(vector, unit, frame)
 
-    assert position_1 == position_1
-    assert position_2 == position_2
-    assert position_1 != position_2
+        assert position is not None
+        assert isinstance(position, Position)
+        assert position.is_defined()
 
+        # Construction with meters static constructor
+        position: Position = Position.meters(vector, frame)
 
-def test_coordinate_position_is_defined():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
-    vector = [1000.0, 0.0, 0.0]
-    position: Position = Position(vector, unit, frame)
+        assert position is not None
+        assert isinstance(position, Position)
+        assert position.is_defined()
 
-    assert position.is_defined()
-    assert Position.undefined().is_defined() is False
+        # Construction with undefined static constructor
+        position: Position = Position.undefined()
 
+        assert position is not None
+        assert isinstance(position, Position)
+        assert position.is_defined() is False
 
-@pytest.mark.skip
-def test_coordinate_position_is_near():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
+    def test_comparators(self, unit: Length.Unit, frame: Frame):
+        vector_1 = [1000.0, 0.0, 0.0]
+        vector_2 = [-450.3, 234.9, -23.0]
 
-    vector_1 = [1000.0, 0.0, 0.0]
-    vector_2 = [1000.00001, 0.0001, 0.0]
+        position_1: Position = Position(vector_1, unit, frame)
+        position_2: Position = Position(vector_2, unit, frame)
 
-    position_1: Position = Position(vector_1, unit, frame)
-    position_2: Position = Position(vector_2, unit, frame)
+        assert position_1 == position_1
+        assert position_2 == position_2
+        assert position_1 != position_2
 
-    assert position_1.is_near(position_2, Unit.Meter)
-    assert position_2.is_near(position_1, Unit.Meter)
+    def test_is_defined(self, unit: Length.Unit, frame: Frame):
+        vector = [1000.0, 0.0, 0.0]
+        position: Position = Position(vector, unit, frame)
 
+        assert position.is_defined()
+        assert Position.undefined().is_defined() is False
 
-def test_coordinate_position_access_frame():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
-    vector = [1000.0, 0.0, 0.0]
+    @pytest.mark.skip
+    def test_is_near(self, unit: Length.Unit, frame: Frame):
 
-    position: Position = Position(vector, unit, frame)
+        vector_1 = [1000.0, 0.0, 0.0]
+        vector_2 = [1000.00001, 0.0001, 0.0]
 
-    ans_frame = position.access_frame()
+        position_1: Position = Position(vector_1, unit, frame)
+        position_2: Position = Position(vector_2, unit, frame)
 
-    assert ans_frame is not None
-    assert isinstance(ans_frame, Frame)
-    assert ans_frame == frame
+        assert position_1.is_near(position_2, Length.Unit.Meter)
+        assert position_2.is_near(position_1, Length.Unit.Meter)
 
+    def test_access_frame(self, unit: Length.Unit, frame: Frame):
+        vector = [1000.0, 0.0, 0.0]
 
-def test_coordinate_position_get_coordinates():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
-    vector = [1000.0, 0.0, 0.0]
+        position: Position = Position(vector, unit, frame)
 
-    position: Position = Position(vector, unit, frame)
+        ans_frame = position.access_frame()
 
-    coordinates = position.get_coordinates()
+        assert ans_frame is not None
+        assert isinstance(ans_frame, Frame)
+        assert ans_frame == frame
 
-    assert coordinates is not None
-    assert isinstance(coordinates, np.ndarray)
-    assert np.array_equal(coordinates, vector)
+    def test_get_coordinates(self, unit: Length.Unit, frame: Frame):
+        vector = [1000.0, 0.0, 0.0]
 
+        position: Position = Position(vector, unit, frame)
 
-def test_coordinate_position_get_unit():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
-    vector = [1000.0, 0.0, 0.0]
+        coordinates = position.get_coordinates()
 
-    position: Position = Position(vector, unit, frame)
+        assert coordinates is not None
+        assert isinstance(coordinates, np.ndarray)
+        assert np.array_equal(coordinates, vector)
 
-    ans_unit = position.get_unit()
+    def test_get_unit(self, unit: Length.Unit, frame: Frame):
+        vector = [1000.0, 0.0, 0.0]
 
-    assert ans_unit is not None
-    assert isinstance(ans_unit, Unit)
-    assert ans_unit == unit
+        position: Position = Position(vector, unit, frame)
 
+        ans_unit = position.get_unit()
 
-def test_coordinate_position_in_unit():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
-    vector = [1000.0, 0.0, 0.0]
+        assert ans_unit is not None
+        assert isinstance(ans_unit, Length.Unit)
+        assert ans_unit == unit
 
-    position: Position = Position(vector, unit, frame)
+    def test_in_unit(self, unit: Length.Unit, frame: Frame):
+        vector = [1000.0, 0.0, 0.0]
 
-    new_position: Position = position.in_unit(Unit.Meter)
+        position: Position = Position(vector, unit, frame)
 
-    assert new_position is not None
-    assert isinstance(new_position, Position)
-    assert new_position == position
+        new_position: Position = position.in_unit(Length.Unit.Meter)
 
-    new_position: Position = position.in_unit(Unit.TerrestrialMile)
+        assert new_position is not None
+        assert isinstance(new_position, Position)
+        assert new_position == position
 
-    assert new_position is not None
-    assert isinstance(new_position, Position)
-    # assert new_position == position
+        new_position: Position = position.in_unit(Length.Unit.TerrestrialMile)
 
+        assert new_position is not None
+        assert isinstance(new_position, Position)
+        # assert new_position == position
 
-def test_coordinate_position_in_meters():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
-    vector = [1000.0, 0.0, 0.0]
+    def test_in_meters(self, unit: Length.Unit, frame: Frame):
+        vector = [1000.0, 0.0, 0.0]
 
-    position: Position = Position(vector, unit, frame)
+        position: Position = Position(vector, unit, frame)
 
-    new_position: Position = position.in_meters()
+        new_position: Position = position.in_meters()
 
-    assert new_position is not None
-    assert isinstance(new_position, Position)
-    assert new_position == position
+        assert new_position is not None
+        assert isinstance(new_position, Position)
+        assert new_position == position
 
+    def test_in_frame(self, unit: Length.Unit, frame: Frame):
+        vector = [1000.0, 0.0, 0.0]
 
-def test_coordinate_position_in_frame():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
-    vector = [1000.0, 0.0, 0.0]
+        instant: Instant = Instant.date_time(DateTime(2020, 1, 1, 0, 0, 0), Scale.UTC)
 
-    instant: Instant = Instant.date_time(DateTime(2020, 1, 1, 0, 0, 0), Scale.UTC)
+        position: Position = Position(vector, unit, frame)
 
-    position: Position = Position(vector, unit, frame)
+        new_position: Position = position.in_frame(frame, instant)
 
-    new_position: Position = position.in_frame(frame, instant)
+        assert new_position is not None
+        assert isinstance(new_position, Position)
+        assert new_position == position
 
-    assert new_position is not None
-    assert isinstance(new_position, Position)
-    assert new_position == position
+    def test_to_string(self, unit: Length.Unit, frame: Frame):
+        vector = [1000.0, 0.0, 0.0]
 
+        position: Position = Position(vector, unit, frame)
 
-def test_coordinate_position_to_string():
-    frame: Frame = Frame.GCRF()
-    unit: Unit = Unit.Meter
-    vector = [1000.0, 0.0, 0.0]
+        string: String = position.to_string()
 
-    position: Position = Position(vector, unit, frame)
+        assert string is not None
+        assert isinstance(string, String)
+        assert string == "[1000.0, 0.0, 0.0] [m] @ GCRF"
 
-    string: String = position.to_string()
+    def test_from_lla(self, lla: LLA, earth):
+        assert Position.from_lla(lla, earth) is not None
 
-    assert string is not None
-    assert isinstance(string, String)
-    assert string == "[1000.0, 0.0, 0.0] [m] @ GCRF"
+        Environment.default(True)
+        assert Position.from_lla(lla) is not None

ostk/physics/test/data/conftest.py

@@ -14,14 +14,17 @@ from ostk.physics.data import Manager, Manifest
 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()
@@ -29,4 +32,3 @@ def manager() -> Manager:
     yield manager
 
     manager.reset()
-

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_manifest.py

@@ -11,6 +11,7 @@ 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
@@ -22,7 +23,9 @@ class TestManifest:
         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)
+        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")

ostk/physics/test/data/test_manifest_manager.py

@@ -12,6 +12,7 @@ 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)
@@ -19,8 +20,8 @@ class TestManager:
     def test_get_remote_url_success(self, manager: Manager):
         assert isinstance(manager.get_remote_url(), URL)
 
-    def test_get_manifest_repository_success(self, manager: Manager):
-        assert isinstance(manager.get_manifest_repository(), Directory)
+    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)
@@ -32,10 +33,10 @@ class TestManager:
         manager.set_remote_url(URL.parse("http://example.com"))
         assert manager.get_remote_url() == URL.parse("http://example.com")
 
-    def test_set_manifest_repository_success(self, manager: Manager):
+    def test_set_local_repository_success(self, manager: Manager):
         directory = Directory.path(Path.parse("./test_directory"))
-        manager.set_manifest_repository(directory)
-        assert manager.get_manifest_repository() == 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)
@@ -59,6 +60,3 @@ class TestManager:
 
     def test_default_remote_url_success(self, manager: Manager):
         assert isinstance(manager.default_remote_url(), URL)
-
-    def test_default_manifest_repository_success(self, manager: Manager):
-        assert isinstance(manager.default_manifest_repository(), Directory)

ostk/physics/test/data/test_scalar.py

@@ -28,7 +28,7 @@ def scalar(value: float, unit: Unit) -> Scalar:
 class TestScalar:
     def test_constructor(self, scalar: Scalar):
         assert scalar is not None
-        
+
     def test_equals(self, scalar: Scalar):
         assert scalar == scalar
 
@@ -49,6 +49,6 @@ class TestScalar:
 
     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

@@ -38,7 +38,7 @@ def vector(value: float, unit: Unit, frame: Frame) -> Vector:
 class TestVector:
     def test_constructor(self, vector: Vector):
         assert vector is not None
-        
+
     def test_equals(self, vector: Vector):
         assert vector == vector
 
@@ -65,6 +65,6 @@ class TestVector:
 
     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/atmospheric/earth/test_manager.py

@@ -4,14 +4,11 @@ import pytest
 
 from datetime import datetime
 import pathlib
-import os
 
 from ostk.core.filesystem import Path
 from ostk.core.filesystem import File
 from ostk.core.filesystem import Directory
 
-from ostk.io import URL
-
 from ostk.physics.time import Scale
 from ostk.physics.time import Instant
 from ostk.physics.time import Duration
@@ -174,14 +171,3 @@ class TestManager:
 
     def test_get_success(self):
         assert isinstance(Manager.get(), Manager)
-
-    def test_default_mode_success(self, manager: Manager):
-        assert manager.default_mode() == Manager.Mode.Automatic
-
-    def test_default_local_repository_success(self, manager: Manager):
-        assert isinstance(manager.default_local_repository(), Directory)
-        assert len(str(manager.default_local_repository().to_string())) > 0
-
-    def test_default_local_repository_lock_timeout_success(self, manager: Manager):
-        assert isinstance(manager.default_local_repository_lock_timeout(), Duration)
-        assert manager.default_local_repository_lock_timeout().in_seconds() == 60.0

ostk/physics/test/environment/gravitational/earth/test_manager.py

@@ -2,12 +2,8 @@
 
 import pytest
 
-import os
-
 from ostk.core.filesystem import Path, Directory, PermissionSet
 
-from ostk.io import URL
-
 from ostk.physics.time import Duration
 from ostk.physics.environment.gravitational import Earth as EarthGravitationalModel
 from ostk.physics.environment.gravitational.earth import (
@@ -19,10 +15,6 @@ from ostk.physics.environment.gravitational.earth import (
 def manager() -> EarthGravitationalModelManager:
     manager = EarthGravitationalModelManager.get()
 
-    manager.set_local_repository(
-        EarthGravitationalModelManager.default_local_repository()
-    )
-
     yield manager
 
     # This reset is important so that other test modules are not affected
@@ -82,20 +74,3 @@ class TestManager:
             manager.get_local_repository().to_string()
             == "./.open-space-toolkit/physics/environment/gravitational/earth2"
         )
-
-    def test_default_local_repository_success(
-        self, manager: EarthGravitationalModelManager
-    ):
-        assert isinstance(
-            EarthGravitationalModelManager.default_local_repository(), Directory
-        )
-        assert len(str(manager.default_local_repository().to_string())) > 0
-
-    def test_default_mode_success(self, manager: EarthGravitationalModelManager):
-        assert manager.default_mode() == EarthGravitationalModelManager.Mode.Automatic
-
-    def test_default_local_repository_lock_timeout_success(
-        self, manager: EarthGravitationalModelManager
-    ):
-        assert isinstance(manager.default_local_repository_lock_timeout(), Duration)
-        assert manager.default_local_repository_lock_timeout().in_seconds() == 60.0

ostk/physics/test/environment/gravitational/test_earth.py

@@ -55,6 +55,14 @@ class TestEarth:
         assert isinstance(earth_gravitational_model, EarthGravitationalModel)
         assert isinstance(earth_gravitational_model, GravitationalModel)
 
+    def test_constructor_success_without_directory_with_degree_and_order(self):
+        earth_gravitational_model = EarthGravitationalModel(
+            EarthGravitationalModel.Type.EGM2008, 2, 2
+        )
+
+        assert isinstance(earth_gravitational_model, EarthGravitationalModel)
+        assert isinstance(earth_gravitational_model, GravitationalModel)
+
     def test_get_type_success(self, earth_gravitational_model: EarthGravitationalModel):
         assert (
             earth_gravitational_model.get_type() == EarthGravitationalModel.Type.EGM2008

ostk/physics/test/environment/magnetic/earth/test_manager.py

@@ -2,13 +2,8 @@
 
 import pytest
 
-import os
-
 from ostk.core.filesystem import Path, Directory, PermissionSet
 
-from ostk.io import URL
-
-from ostk.physics.time import Duration
 from ostk.physics.environment.magnetic import Earth as EarthMagneticModel
 from ostk.physics.environment.magnetic.earth import (
     Manager as EarthMagneticModelManager,
@@ -19,8 +14,6 @@ from ostk.physics.environment.magnetic.earth import (
 def manager() -> EarthMagneticModelManager:
     manager = EarthMagneticModelManager.get()
 
-    manager.set_local_repository(EarthMagneticModelManager.default_local_repository())
-
     yield manager
 
     # This reset is important so that other test modules are not affected
@@ -69,16 +62,3 @@ class TestManager:
             manager.get_local_repository().to_string()
             == "./.open-space-toolkit/physics/environment/magnetic/earth2"
         )
-
-    def test_default_local_repository_success(self, manager: EarthMagneticModelManager):
-        assert isinstance(EarthMagneticModelManager.default_local_repository(), Directory)
-        assert len(str(manager.default_local_repository().to_string())) > 0
-
-    def test_default_mode_success(self, manager: EarthMagneticModelManager):
-        assert manager.default_mode() == EarthMagneticModelManager.Mode.Automatic
-
-    def test_default_local_repository_lock_timeout_success(
-        self, manager: EarthMagneticModelManager
-    ):
-        assert isinstance(manager.default_local_repository_lock_timeout(), Duration)
-        assert manager.default_local_repository_lock_timeout().in_seconds() == 60.0