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

{open_space_toolkit_astrodynamics-5.1.5.dist-info → open_space_toolkit_astrodynamics-5.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: open-space-toolkit-astrodynamics
-Version: 5.1.5
+Version: 5.2.0
 Summary: Orbit, attitude, access.
 Author: Open Space Collective
 Author-email: contact@open-space-collective.org

open_space_toolkit_astrodynamics-5.2.0.dist-info/RECORD

@@ -0,0 +1,96 @@
+ostk/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/OpenSpaceToolkitAstrodynamicsPy.cpython-38-x86_64-linux-gnu.so,sha256=VvN9bLMp87RyeNcZS0dTduOx0P6naFrxcyCAX4z6gh0,2004760
+ostk/astrodynamics/__init__.py,sha256=3gWyqFIbhAfcdeMhmfBPQPlPQTmaOzm-6flkJe745Zk,251
+ostk/astrodynamics/converters.py,sha256=IUxJK5qNzDMsqMSiMT6hUThCjncKoIMW0ifu8kwbww0,4680
+ostk/astrodynamics/display.py,sha256=y9FnoQbPFGM6LzkUdgXgeqtuVGhv57GuTKbeDdcFPgw,6306
+ostk/astrodynamics/libopen-space-toolkit-astrodynamics.so.5,sha256=7okoTSc8oHuKbfLThcRJXkgb7EuYdzR4awMUcGsfNgM,94498664
+ostk/astrodynamics/utilities.py,sha256=NbeKN_CtcCRHIinhoNWTEosBMR7xc924S-ZFD97mduw,4478
+ostk/astrodynamics/viewer.py,sha256=B5m-eNWQrd4vIBqmT7-400IfHldxU5lFDTLpfHiiHj4,8910
+ostk/astrodynamics/pytrajectory/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/pytrajectory/pystate.py,sha256=ceNZWYCqtsXWJlE6JKDxB5qn5ixTWGOwrYXge21XPmk,1037
+ostk/astrodynamics/test/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/conftest.py,sha256=J3SbQibnMVaPktAF6Z6cwVCa7xhW3uXpkOJRom6YtCo,2719
+ostk/astrodynamics/test/test_access.py,sha256=6sWINwQw-KMKKOfuJIJxyIIrmT513VjvoXt2rNPpm0w,3959
+ostk/astrodynamics/test/test_converters.py,sha256=hnvqOCtUGrzBvjv7GnmaRdl2mlKM3eWuJUspe305B-A,12114
+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=i164XnG6n-_FteiO6rjXA7GGsJhMdmT-508DpKaveVY,3957
+ostk/astrodynamics/test/access/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/access/test_generator.py,sha256=6cwC5GGvnWLxPXmgRQ-N7tjorGUfPxA8qFcdGaTmpp4,8038
+ostk/astrodynamics/test/conjunction/messages/ccsds/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/conjunction/messages/ccsds/conftest.py,sha256=xyTd24828Ue2HgOWY8UhizIc5mzgRgoH2VSOfVfKCdw,14632
+ostk/astrodynamics/test/conjunction/messages/ccsds/test_cdm.py,sha256=SqtYXB8qhKzbQEWmCzz13xi0Uu64-VSKXuMObcFV7EI,18518
+ostk/astrodynamics/test/conjunction/messages/ccsds/data/cdm.json,sha256=-CL8Bu-ivigzkrqHs9Q5snSubHJ_D0YMroatEInkidc,12965
+ostk/astrodynamics/test/dynamics/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/dynamics/test_atmospheric_drag.py,sha256=mKXF1r-gaRz7vtMNULsl6onFOwD3uk0lYqxKStm4bxU,3765
+ostk/astrodynamics/test/dynamics/test_central_body_gravity.py,sha256=x2_Xe-nGVkhKuq7p5Wuq7exeXKg_de3sAGrwGsOSRfA,1790
+ostk/astrodynamics/test/dynamics/test_dynamics.py,sha256=FU5Qc3H6wkguml1Kt_hTX8uKplEyfcbH8wkC1TT4bjE,1240
+ostk/astrodynamics/test/dynamics/test_position_derivative.py,sha256=HIHH8W3WR0eHJo82N6k03CBj_i1g2OuiHm-xEumPbx4,1618
+ostk/astrodynamics/test/dynamics/test_third_body_gravity.py,sha256=hl3svhU6gNJdKoqqdgsZt-Ngrex9whcuLPeCgepM4G8,1887
+ostk/astrodynamics/test/dynamics/test_thruster.py,sha256=HHFei_ZdUcwdRNXcKDTF-EhYe8ZjVht-kQjUl0nhILU,4403
+ostk/astrodynamics/test/event_condition/test_angular_condition.py,sha256=OFlEbUDWaUs1kjUSf0AhIsOKV5qSUj6_sOxLQK5q8AU,3120
+ostk/astrodynamics/test/event_condition/test_boolean_condition.py,sha256=Ng6FLKD8XoUHUz6-gi3wvrYhSGNSB_iJ_n89ajN4O3s,1580
+ostk/astrodynamics/test/event_condition/test_coe_condition.py,sha256=1znTEcKVhR54CVfjhQeeAxbe4CdJJ1qlsKc8TK9jwqw,2581
+ostk/astrodynamics/test/event_condition/test_instant_condition.py,sha256=Qr5dN6nrK7XDff7L4vnGCqasMK-Fa6qvxwy_U0xgH5M,1438
+ostk/astrodynamics/test/event_condition/test_logical_condition.py,sha256=09h5TYWtwGt4NQW3k_tziiVs0Q2981rNII9wyKg7p2Q,3321
+ostk/astrodynamics/test/event_condition/test_real_condition.py,sha256=tle6HVzMFMIIkfRY7CuaA0mPtw3riJBG_JQkc1L0dpk,1374
+ostk/astrodynamics/test/flight/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/flight/test_profile.py,sha256=OBwXlN35Fosk9ue37cDl6NAdITv8JG3s7mnHq3_cxc8,5195
+ostk/astrodynamics/test/flight/test_system.py,sha256=YPrjoV3o67_1zCXCZqohlSAR7zkU0LXP2fF20M2s0C8,1326
+ostk/astrodynamics/test/flight/profile/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/flight/profile/test_state.py,sha256=2vc-PHFr9M9_x08n0ZCm29O4mE-zeu_0oTW8Wd4x7KY,4266
+ostk/astrodynamics/test/flight/system/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/flight/system/test_propulsion_system.py,sha256=ei-25MZczwzhiV5QVB9LXdmBY3VnBHr1DEVn_Zqtb1A,1027
+ostk/astrodynamics/test/flight/system/test_satellite_system.py,sha256=rPka530ZqX_XBVmt0SC5o765O1O90_m_U5cEUsrQuWU,2530
+ostk/astrodynamics/test/flight/system/test_satellite_system_builder.py,sha256=G3MUD-mBTakr-ip_261kwBTvTclnAArPPeHY3ZF9PYM,2430
+ostk/astrodynamics/test/guidance_law/test_constant_thrust.py,sha256=YpMkMvat4YnE1Fyy5vZ1Sf3VvbvDaC_UvbIKNNCG_B4,2635
+ostk/astrodynamics/test/guidance_law/test_qlaw.py,sha256=nu56zm7D9uwonCXCjpNw49nv9TApelTSlsoItQQ6VhY,3677
+ostk/astrodynamics/test/solvers/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/solvers/test_finite_difference_solver.py,sha256=h_CDEswTKzEld9vvTMMPXj-TI3YSeY14-Z6clYWtfB0,5417
+ostk/astrodynamics/test/solvers/test_temporal_condition_solver.py,sha256=s7r7VelXx1_b-6wO_GyOlmxlsfA4I_zpxrWojriQls8,4867
+ostk/astrodynamics/test/trajectory/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/test_local_orbital_frame_direction.py,sha256=d72J50UGG9m0Atei0UQ8sITU1WdJmws5xgUiacLZMbw,2515
+ostk/astrodynamics/test/trajectory/test_local_orbital_frame_factory.py,sha256=FziYh9XxD2FEuj6WhnqOe5mF9twLTQtvcem8KzQZKXg,1864
+ostk/astrodynamics/test/trajectory/test_model.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/test_orbit.py,sha256=SlCBZ_broPaJ6L8khMuvbdPem9-zmqG6IZwxr9cECB4,3155
+ostk/astrodynamics/test/trajectory/test_propagator.py,sha256=kEAvrFM4MZouAAnYPi91akPSXWHbGxHTo5uaXQKexII,12631
+ostk/astrodynamics/test/trajectory/test_segment.py,sha256=UxSyasdBQoaT7bTBlRoBxhQ3paHw1JRDR7GJzFTmf6M,8731
+ostk/astrodynamics/test/trajectory/test_sequence.py,sha256=uZdHNY7wAvkSMQEzNId3DWQOilsKdCEbqBd-AbqMYVE,12412
+ostk/astrodynamics/test/trajectory/test_state.py,sha256=rW0HH7oy_tZ-34VIhk2rx_v2Kyg3xegBYpKs9_NznSs,6508
+ostk/astrodynamics/test/trajectory/test_state_builder.py,sha256=3C8JeN8VGgUzEASgh5PfnJQ2JWGkKYTm7eaelE6Lw7E,4815
+ostk/astrodynamics/test/trajectory/orbit/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/orbit/test_model.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/orbit/test_pass.py,sha256=MtPRKhVz4hCzfFuE0p1x89Ps68M9s7miT7VtPIDVeOo,1898
+ostk/astrodynamics/test/trajectory/orbit/messages/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/orbit/messages/spacex/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/orbit/messages/spacex/conftest.py,sha256=O75ywkPJjoFp0uKptCu4C2tGMRaGN-8NVFVhps85eWE,379
+ostk/astrodynamics/test/trajectory/orbit/messages/spacex/test_opm.py,sha256=wfxuccqIF2odRA9Dc-8wtKZ8t__4KJ9HVhN-il284OU,3893
+ostk/astrodynamics/test/trajectory/orbit/messages/spacex/data/opm_1.yaml,sha256=DqIdfLCPD9oV4qDvgq35fIT8Gb8zgtrHr5Nx53InSQM,1363
+ostk/astrodynamics/test/trajectory/orbit/models/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/orbit/models/test_kepler.py,sha256=sf1FpdYyiGyWcal-CwVfelW1k-HYGlIdbJdcbOi9-lY,3372
+ostk/astrodynamics/test/trajectory/orbit/models/test_propagated.py,sha256=PnzJsRVn-TKQ7fcsmxm4YYE4e2Tr1xvBTB1r0RboRyk,6678
+ostk/astrodynamics/test/trajectory/orbit/models/test_sgp4.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/orbit/models/test_tabulated.py,sha256=tL3gGDitdEYCCRwGtu2RP-9d4JWsujV53ck9DRkyeG4,11183
+ostk/astrodynamics/test/trajectory/orbit/models/kepler/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_brouwer_lyddane_mean.py,sha256=9JC4LTpMXWzQ2VoEOaVaruy2hPjyiAIR5_2QTfm4MqU,1545
+ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_brouwer_lyddane_mean_long.py,sha256=gVpnA6FE3aCD6hskOKiVkb4YlONTUMogwBY1EaOqm7M,2472
+ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_brouwer_lyddane_mean_short.py,sha256=8g2Q8ywuU5wsvrycAKUWFY-1bqIcylBpcnMJW4ze8Hs,2489
+ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_coe.py,sha256=C_2qGt8yzbs6S-rCheUbj8wOnLEWBNio2UD6ST-SwHk,4781
+ostk/astrodynamics/test/trajectory/orbit/models/sgp4/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
+ostk/astrodynamics/test/trajectory/orbit/models/sgp4/test_tle.py,sha256=lbPqHd87ExTNbBFcFSTMWRnWbpe_4BDJ2g2APlQLJkY,11486
+ostk/astrodynamics/test/trajectory/state/test_coordinates_broker.py,sha256=LUVQ6LjpWAxs6XA3vOwHO5Unnf4TMYCx31fk_ORLTcc,2907
+ostk/astrodynamics/test/trajectory/state/test_coordinates_subset.py,sha256=eMmifR5rJMk2yFN7pfAHq2Ub3FPvknb6gvS-nBDidro,1201
+ostk/astrodynamics/test/trajectory/state/test_numerical_solver.py,sha256=-SHQWgR34Xk0gO26VwQWbmRnPXp3AKfk0j9t-lPvfmk,9403
+ostk/astrodynamics/test/trajectory/state/coordinates_subset/test_angular_velocity.py,sha256=IWLmoMvxVejdiw8JuiDsIaii7ctbRDvU-13fgZEvWXA,626
+ostk/astrodynamics/test/trajectory/state/coordinates_subset/test_attitude_quaternion.py,sha256=B1Tni_YpKB9Fzb0fpXSokw9DjgiZem9uBATMQbcq8P4,394
+ostk/astrodynamics/test/trajectory/state/coordinates_subset/test_cartesian_position.py,sha256=L3MpDG9MDHbK2HPZejqiY0QJdiEB-FZe0rVGDl379YA,2722
+ostk/astrodynamics/test/trajectory/state/coordinates_subset/test_cartesian_velocity.py,sha256=k0aNqdMwwt-XSFQ1Si8kDTY79SB7fC0WWm2UNjpaOI0,3077
+open_space_toolkit_astrodynamics-5.2.0.dist-info/METADATA,sha256=R5UAtA3PdcwUnrltBkZOUeU1DQuv8Ggg-iY0Pfy7Kc0,1777
+open_space_toolkit_astrodynamics-5.2.0.dist-info/WHEEL,sha256=LNQH4F1F9_Ua4GgazGQDFGQhQ8B1tSrkPIuEzIRv9nc,93
+open_space_toolkit_astrodynamics-5.2.0.dist-info/top_level.txt,sha256=zOR18699uDYnafgarhL8WU_LmTZY_5NVqutv-flp_x4,5
+open_space_toolkit_astrodynamics-5.2.0.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+open_space_toolkit_astrodynamics-5.2.0.dist-info/RECORD,,

ostk/__init__.py

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

ostk/astrodynamics/converters.py

@@ -15,12 +15,12 @@ from ostk.physics.coordinate import Velocity
 from ostk.physics.coordinate import Frame
 
 
-def coerce_to_datetime(value: Instant | datetime) -> datetime:
+def coerce_to_datetime(value: Instant | datetime | str) -> datetime:
     """
     Return datetime from value.
 
     Args:
-        value (Instant | datetime): A value to coerce.
+        value (Instant | datetime | str): A value to coerce.
 
     Returns:
         datetime: The coerced datetime.
@@ -29,15 +29,21 @@ def coerce_to_datetime(value: Instant | datetime) -> datetime:
     if isinstance(value, datetime):
         return value
 
-    return value.get_date_time(Scale.UTC).replace(tzinfo=timezone.utc)
+    if isinstance(value, Instant):
+        return value.get_date_time(Scale.UTC).replace(tzinfo=timezone.utc)
+
+    if isinstance(value, str):
+        return datetime.fromisoformat(value)
+
+    raise TypeError("Argument must be a datetime, an Instant, or a str.")
 
 
-def coerce_to_instant(value: Instant | datetime) -> Instant:
+def coerce_to_instant(value: Instant | datetime | str) -> Instant:
     """
     Return Instant from value.
 
     Args:
-        value (Instant | datetime): A value to coerce.
+        value (Instant | datetime | str): A value to coerce.
 
     Returns:
         Instant: The coerced Instant.
@@ -46,7 +52,39 @@ def coerce_to_instant(value: Instant | datetime) -> Instant:
     if isinstance(value, Instant):
         return value
 
-    return Instant.date_time(value, Scale.UTC)
+    if isinstance(value, datetime):
+        return Instant.date_time(value.astimezone(tz=timezone.utc), Scale.UTC)
+
+    if isinstance(value, str):
+        return coerce_to_instant(coerce_to_datetime(value))
+
+    raise TypeError("Argument must be a datetime, an Instant, or a str.")
+
+
+def coerce_to_iso(
+    value: Instant | datetime | str, timespec: str = "microseconds"
+) -> Instant:
+    """
+    Return an ISO string from value.
+
+    Args:
+        value (Instant | datetime | str): A value to coerce.
+        timespec (str): A time resolution. Defaults to "microseconds".
+
+    Returns:
+        str: The coerced ISO string.
+    """
+
+    if isinstance(value, str):
+        return coerce_to_iso(coerce_to_datetime(value), timespec=timespec)
+
+    if isinstance(value, datetime):
+        return value.isoformat(timespec=timespec)
+
+    if isinstance(value, Instant):
+        return coerce_to_iso(coerce_to_datetime(value), timespec=timespec)
+
+    raise TypeError("Argument must be a datetime, an Instant, or a str.")
 
 
 def coerce_to_interval(

ostk/astrodynamics/pytrajectory/pystate.py

@@ -28,9 +28,7 @@ def custom_class_generator(frame: Frame, coordinates_subsets: list) -> type:
         state_builder: StateBuilder = StateBuilder(frame, coordinates_subsets)
 
         def __init__(self, instant: Instant, coordinates: np.ndarray):
-            super().__init__(
-                StateTemplateType.state_builder.build(instant, coordinates)
-            )
+            super().__init__(StateTemplateType.state_builder.build(instant, coordinates))
 
     return StateTemplateType
 

ostk/astrodynamics/test/access/__init__.py

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

ostk/astrodynamics/test/access/test_generator.py

@@ -0,0 +1,248 @@
+# Apache License 2.0
+
+import pytest
+
+from ostk.mathematics.objects import RealInterval
+
+from ostk.physics.units import Length
+from ostk.physics.units import Angle
+from ostk.physics.time import DateTime
+from ostk.physics.time import Scale
+from ostk.physics.time import Duration
+from ostk.physics.time import Instant
+from ostk.physics.time import Interval
+from ostk.physics import Environment
+from ostk.physics.environment.objects import Celestial
+
+from ostk.astrodynamics import Trajectory
+from ostk.astrodynamics.trajectory import Orbit
+from ostk.astrodynamics.trajectory.orbit.models import Kepler
+from ostk.astrodynamics.trajectory.orbit.models.kepler import COE
+from ostk.astrodynamics import Access
+from ostk.astrodynamics.access import Generator
+
+
+@pytest.fixture
+def environment() -> Environment:
+    return Environment.default()
+
+
+@pytest.fixture
+def earth(environment: Environment) -> Celestial:
+    return environment.access_celestial_object_with_name("Earth")
+
+
+@pytest.fixture
+def generator(environment: Environment) -> Generator:
+    return Generator(
+        environment=environment,
+        aer_filter=lambda aer: True,
+        access_filter=lambda access: True,
+        state_filter=lambda state_1, state_2: True,
+    )
+
+
+@pytest.fixture
+def from_trajectory(earth: Celestial) -> Trajectory:
+    return Orbit(
+        model=Kepler(
+            coe=COE(
+                semi_major_axis=Length.kilometers(7000.0),
+                eccentricity=0.0,
+                inclination=Angle.degrees(45.0),
+                raan=Angle.degrees(0.0),
+                aop=Angle.degrees(0.0),
+                true_anomaly=Angle.degrees(0.0),
+            ),
+            epoch=Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC),
+            celestial_object=earth,
+            perturbation_type=Kepler.PerturbationType.No,
+        ),
+        celestial_object=earth,
+    )
+
+
+@pytest.fixture
+def to_trajectory(earth: Celestial) -> Trajectory:
+    return Orbit(
+        model=Kepler(
+            coe=COE(
+                semi_major_axis=Length.kilometers(7000.0),
+                eccentricity=0.0,
+                inclination=Angle.degrees(45.0),
+                raan=Angle.degrees(180.0),
+                aop=Angle.degrees(0.0),
+                true_anomaly=Angle.degrees(180.0),
+            ),
+            epoch=Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC),
+            celestial_object=earth,
+            perturbation_type=Kepler.PerturbationType.No,
+        ),
+        celestial_object=earth,
+    )
+
+
+class TestGenerator:
+    def test_constructor_success_environment(self, environment: Environment):
+        generator = Generator(
+            environment=environment,
+        )
+
+        assert generator is not None
+        assert isinstance(generator, Generator)
+
+    def test_constructor_success_environment_aer_filter(self, environment: Environment):
+        generator = Generator(
+            environment=environment,
+            aer_filter=lambda aer: True,
+        )
+
+        assert generator is not None
+        assert isinstance(generator, Generator)
+
+    def test_constructor_success_environment_access_filter(
+        self,
+        environment: Environment,
+    ):
+        generator = Generator(
+            environment=environment,
+            access_filter=lambda access: True,
+        )
+
+        assert generator is not None
+        assert isinstance(generator, Generator)
+
+    def test_constructor_success_environment_state_filter(
+        self,
+        environment: Environment,
+    ):
+        generator = Generator(
+            environment=environment,
+            state_filter=lambda state_1, state_2: True,
+        )
+
+        assert generator is not None
+        assert isinstance(generator, Generator)
+
+    def test_constructor_success_environment_step_tolerance(
+        self,
+        environment: Environment,
+    ):
+        generator = Generator(
+            environment=environment,
+            step=Duration.seconds(1.0),
+            tolerance=Duration.minutes(1.0),
+        )
+
+        assert generator is not None
+        assert isinstance(generator, Generator)
+        assert generator.get_step() == Duration.seconds(1.0)
+        assert generator.get_tolerance() == Duration.minutes(1.0)
+
+    def test_getters_success(self, generator: Generator):
+        assert generator.get_step() == Duration.minutes(1.0)
+        assert generator.get_tolerance() == Duration.microseconds(1.0)
+        assert generator.get_aer_filter() is not None
+        assert generator.get_access_filter() is not None
+        assert generator.get_state_filter() is not None
+
+    def test_get_condition_function_success(
+        self,
+        generator: Generator,
+        from_trajectory: Trajectory,
+        to_trajectory: Trajectory,
+    ):
+        condition_function = generator.get_condition_function(
+            from_trajectory=from_trajectory,
+            to_trajectory=to_trajectory,
+        )
+
+        assert condition_function is not None
+        assert (
+            condition_function(
+                Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC)
+            )
+            is True
+        )
+
+    def test_compute_accesses_success(
+        self,
+        generator: Generator,
+        from_trajectory: Trajectory,
+        to_trajectory: Trajectory,
+    ):
+        accesses = generator.compute_accesses(
+            interval=Interval.closed(
+                Instant.date_time(DateTime(2018, 1, 1, 0, 0, 0), Scale.UTC),
+                Instant.date_time(DateTime(2018, 1, 1, 2, 0, 0), Scale.UTC),
+            ),
+            from_trajectory=from_trajectory,
+            to_trajectory=to_trajectory,
+        )
+
+        assert accesses is not None
+        assert isinstance(accesses, list)
+        assert accesses[0] is not None
+        assert isinstance(accesses[0], Access)
+
+    def test_set_step_success(self, generator: Generator):
+        generator.set_step(Duration.seconds(1.0))
+
+    def test_set_tolerance_success(self, generator: Generator):
+        generator.set_tolerance(Duration.seconds(1.0))
+
+    def test_set_aer_filter_success(self, generator: Generator):
+        generator.set_aer_filter(aer_filter=lambda aer: True)
+
+    def test_set_access_filter_success(self, generator: Generator):
+        generator.set_access_filter(access_filter=lambda access: True)
+
+    def test_set_state_filter_success(self, generator: Generator):
+        generator.set_state_filter(state_filter=lambda state_1, state_2: True)
+
+    def test_undefined_success(self):
+        generator = Generator.undefined()
+
+        assert generator is not None
+        assert isinstance(generator, Generator)
+        assert generator.is_defined() is False
+
+    def test_aer_ranges_success(self, environment: Environment):
+        # Construct arbitrary AER ranges
+        azimuth_interval = RealInterval.closed(0.0, 360.0)
+        elevation_interval = RealInterval.closed(0.0, 90.0)
+        range_interval = RealInterval.closed(0.0, 7000e3)
+
+        generator = Generator.aer_ranges(
+            azimuth_range=azimuth_interval,
+            elevation_range=elevation_interval,
+            range_range=range_interval,
+            environment=environment,
+        )
+
+        assert generator is not None
+        assert isinstance(generator, Generator)
+        assert generator.is_defined()
+
+    def test_aer_mask_success(self, environment: Environment):
+        # Construct arbitrary anAzimuthElevationMask using python dict
+        an_azimuth_elevation_mask = {
+            0.0: 30.0,
+            90.0: 60.0,
+            180.0: 60.0,
+            270.0: 30.0,
+            359.0: 30.0,
+        }
+
+        # Construct arbitrary aRangerange
+        a_range_range = RealInterval(0.0, 10e4, RealInterval.Type.Closed)
+
+        generator = Generator.aer_mask(
+            azimuth_elevation_mask=an_azimuth_elevation_mask,
+            range_range=a_range_range,
+            environment=environment,
+        )
+
+        assert generator is not None
+        assert isinstance(generator, Generator)
+        assert generator.is_defined()

ostk/astrodynamics/test/conjunction/messages/ccsds/__init__.py

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