open-space-toolkit-astrodynamics 12.0.1__py310-none-manylinux2014_x86_64.whl → 12.2.0__py310-none-manylinux2014_x86_64.whl

{open_space_toolkit_astrodynamics-12.0.1.dist-info → open_space_toolkit_astrodynamics-12.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: open-space-toolkit-astrodynamics
-Version: 12.0.1
+Version: 12.2.0
 Summary: Orbit, attitude, access.
 Author: Open Space Collective
 Author-email: contact@open-space-collective.org
@@ -11,10 +11,10 @@ Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: POSIX :: Linux
 Description-Content-Type: text/markdown
-Requires-Dist: open-space-toolkit-core ~=4.0
+Requires-Dist: open-space-toolkit-core ~=4.1
 Requires-Dist: open-space-toolkit-io ~=4.0
-Requires-Dist: open-space-toolkit-mathematics ~=4.0
-Requires-Dist: open-space-toolkit-physics ~=9.0
+Requires-Dist: open-space-toolkit-mathematics ~=4.2
+Requires-Dist: open-space-toolkit-physics ~=10.1
 
 # Open Space Toolkit ▸ Astrodynamics
 

{open_space_toolkit_astrodynamics-12.0.1.dist-info → open_space_toolkit_astrodynamics-12.2.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
 ostk/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
-ostk/astrodynamics/OpenSpaceToolkitAstrodynamicsPy.cpython-310-x86_64-linux-gnu.so,sha256=BfVGsn5nhBe4kRaRSNV8kNAYj_bA2CDdHhTPox0pCPI,2422720
+ostk/astrodynamics/OpenSpaceToolkitAstrodynamicsPy.cpython-310-x86_64-linux-gnu.so,sha256=bhOEl5LpzdX_dSO1iBYhvGu_-RoWb5OXaxRtWRKhiFc,2447328
 ostk/astrodynamics/__init__.py,sha256=3gWyqFIbhAfcdeMhmfBPQPlPQTmaOzm-6flkJe745Zk,251
 ostk/astrodynamics/converters.py,sha256=Ld-DLTaqg7pBkIJAcYeoaA-sDOQbR_uksRR-lg9dfXo,3278
 ostk/astrodynamics/dataframe.py,sha256=njimcfT2cvnSOp9eBR1Ej-LuYpXbVieAVEBk2tYW4Vg,18761
 ostk/astrodynamics/display.py,sha256=KiEGsjNftQfoUe-HfZeYpQQQ4TkaEnxtEl6p5_LX5f0,6303
-ostk/astrodynamics/libopen-space-toolkit-astrodynamics.so.12,sha256=gie8faYXnNGJladdNg6RIYnSuZ7HRQ0175QZNZxihCs,84588664
+ostk/astrodynamics/libopen-space-toolkit-astrodynamics.so.12,sha256=jIwo6Hr9NJHfof3i6Or_ZYBhR3Zej8sqy6uQOHMAhR4,87107488
 ostk/astrodynamics/utilities.py,sha256=f0ySFhD6X9jhAZW1hLnbLVEOuy3QgVq2C6k4m8MFGYA,6963
 ostk/astrodynamics/viewer.py,sha256=8UxqsvU5wui7gPUwxXaKcIB-509NJiih9Df3mQgXyXY,11776
 ostk/astrodynamics/pytrajectory/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
-ostk/astrodynamics/pytrajectory/pystate.py,sha256=qtGofajIODveyy_g_lEeBqJQ1GddqmzRBVNCExF8pdk,5734
+ostk/astrodynamics/pytrajectory/pystate.py,sha256=IoR4AIqZe7KuvxrNbnpPtQLC-bY1RPxUlSB8TbfmANg,6681
 ostk/astrodynamics/test/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
-ostk/astrodynamics/test/conftest.py,sha256=fsdUQnLxyfvBzfaSCkgipkPIhkPHchQqbeAro1pF648,2717
+ostk/astrodynamics/test/conftest.py,sha256=stmQOt7UXjBlXKJzNN6RkS2miv1xXSOX-YNpFhaHTqI,2771
 ostk/astrodynamics/test/test_access.py,sha256=MCBsUPtuVm7NgHZR0z0DpWnPZ_qBu4aRhLI2PnRNUYs,3940
 ostk/astrodynamics/test/test_converters.py,sha256=mFpDD0YM8o356lj91GGIwYdMk2ut6xZFV3uYcgZepMY,8744
 ostk/astrodynamics/test/test_dataframe.py,sha256=IJMOODzTVYtiPxekmltH6lOMArPXbHwSEAl-jg0Ab2g,28414
@@ -18,7 +18,7 @@ ostk/astrodynamics/test/test_display.py,sha256=Ykvw2rUWWw8jsX0Suy11X27bHjwm_ira_
 ostk/astrodynamics/test/test_event_condition.py,sha256=mhMTH7wAoYFWRYt_8l2d1vjNPrFhVjMAEET4INLCVXY,1472
 ostk/astrodynamics/test/test_import.py,sha256=py_hALBR0IYuUzv9dfgQZzrrLHJIpnyKvt3Oi1XBqCg,1251
 ostk/astrodynamics/test/test_root_solver.py,sha256=hQ8O6g-WP49gZH_H3Rdufv0F0gQorpzJyIcjBGGUQ34,1831
-ostk/astrodynamics/test/test_trajectory.py,sha256=WC9X-O_prHeVLIZlmmYzgkskrs5ZwvmOfhz8yth2lgI,1345
+ostk/astrodynamics/test/test_trajectory.py,sha256=0sgGPkMFy-u-33z6SF-sTvaBb_NU7OvaeMTVF6wevfY,3148
 ostk/astrodynamics/test/test_utilities.py,sha256=xmRt-5z9-FFYTMQxBpjGrcKP39trbOiH-NLxGcTWOJE,3048
 ostk/astrodynamics/test/test_viewer.py,sha256=rbVAiclzBZ1bS4Zdh4xUNJ-STCN4G0O7Gr8suSPJrSc,3986
 ostk/astrodynamics/test/access/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
@@ -45,7 +45,7 @@ ostk/astrodynamics/test/event_condition/test_logical_condition.py,sha256=09h5TYW
 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_maneuver.py,sha256=6LhnNhaiwId-IveUraHnhx8dYxeOgzntP_WPtv9GFjQ,6197
-ostk/astrodynamics/test/flight/test_profile.py,sha256=ZuK2FWWnFzvnJsMgZYeQyG6G0yolUihL4TjGt0BMrZI,7518
+ostk/astrodynamics/test/flight/test_profile.py,sha256=-MI7mtkvJdq7eFr6terpD-2na306VRzYBTJ36qigjB4,7518
 ostk/astrodynamics/test/flight/test_system.py,sha256=MVaE7lJYisH4vmJPD-G-Hw4wNj-Xe8yMksgu8IXCLvg,1322
 ostk/astrodynamics/test/flight/system/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
 ostk/astrodynamics/test/flight/system/test_propulsion_system.py,sha256=jLUC74YjIQz7APr-kQcdYRYiKm8lfequjosptG29kmI,1964
@@ -64,7 +64,7 @@ ostk/astrodynamics/test/trajectory/test_orbit.py,sha256=iKVgScEo1osMIXjON4ToURCg
 ostk/astrodynamics/test/trajectory/test_propagator.py,sha256=dG31S5FY0NjJ5bnisqJ_HdL7amj_AZAWWLiOmdHfJAc,14233
 ostk/astrodynamics/test/trajectory/test_segment.py,sha256=ACVMYvjlTyCQHZ2-hFEurK08-LQxt6VlcGTMOrgDHaE,9333
 ostk/astrodynamics/test/trajectory/test_sequence.py,sha256=gswWqEBTYvbi5TQjF0DNPkAj5COo9Nb9IK7TdwJ_5C0,12882
-ostk/astrodynamics/test/trajectory/test_state.py,sha256=IRtdEyiYHCdBXSRmNEKZy355s7i8STYddOzQAMXh5EY,10805
+ostk/astrodynamics/test/trajectory/test_state.py,sha256=rgJ4lHgp71axKdvdGSkk0sQbQ9i00gBi8ewLxXCojmo,13316
 ostk/astrodynamics/test/trajectory/test_state_builder.py,sha256=PSlXtmGURCrlf0w50tn3dEBRo6p0x4NmIF9gJsNOR3k,4782
 ostk/astrodynamics/test/trajectory/orbit/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
 ostk/astrodynamics/test/trajectory/orbit/test_model.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
@@ -83,7 +83,7 @@ ostk/astrodynamics/test/trajectory/orbit/models/kepler/__init__.py,sha256=epnVn2
 ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_brouwer_lyddane_mean.py,sha256=NvtwHJzZvfp8LZGcRhWU1kJ_rdLCim8GBXUbzMrPjjc,1542
 ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_brouwer_lyddane_mean_long.py,sha256=c-8M4py0LEDHoRuc-ukmUyGYN7PcSDWFKY2Bh1KFNRk,2468
 ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_brouwer_lyddane_mean_short.py,sha256=rLA6Ud41SFZSWZiOzGFU1FBPlzv6-c6eP8Ly-bqsFKo,2485
-ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_coe.py,sha256=H42bc4SkXl7tvEMDwAdPImiJpFPBRjSD_Npsmxq4xaY,4778
+ostk/astrodynamics/test/trajectory/orbit/models/kepler/test_coe.py,sha256=inXqK6zD3VgxvBZg-bh3AzLTh6q12hxJ8iiAyIodYMs,5319
 ostk/astrodynamics/test/trajectory/orbit/models/sgp4/__init__.py,sha256=epnVn2PwdQkUDZ1msqBRO5nEZIOUBIq-IfK3IlNPijE,21
 ostk/astrodynamics/test/trajectory/orbit/models/sgp4/test_tle.py,sha256=jkyRr5NJQMaSpI3ey7cKcH0E-BlalaUlf_mZPYTSow8,11482
 ostk/astrodynamics/test/trajectory/state/test_coordinate_broker.py,sha256=ho5FBfB3IvIjfQl7EYoIN1KBuH9lu76IiNMuweq8cqg,2844
@@ -93,8 +93,8 @@ ostk/astrodynamics/test/trajectory/state/coordinate_subset/test_angular_velocity
 ostk/astrodynamics/test/trajectory/state/coordinate_subset/test_attitude_quaternion.py,sha256=UEu9ApzQLmT87eeISw6_gcHTlX-4b2scIvHz-uE1p_c,393
 ostk/astrodynamics/test/trajectory/state/coordinate_subset/test_cartesian_position.py,sha256=XvHdk1KjacTwtkgx2jUAc9I9N3nvjPDv03FAanpv8jQ,2702
 ostk/astrodynamics/test/trajectory/state/coordinate_subset/test_cartesian_velocity.py,sha256=-kd5TZO5TICihbkqDTew2i_tDpggdpe3Yf23046FATM,3057
-open_space_toolkit_astrodynamics-12.0.1.dist-info/METADATA,sha256=Uz-7aAVpAbeRD9LZcYvamuARdgSEJa9ZbMi3GzE7Gjk,1916
-open_space_toolkit_astrodynamics-12.0.1.dist-info/WHEEL,sha256=EfzGwk6HazeuFg6dmyxdAa9hqpFBtmJ_haGsqhCVSCQ,110
-open_space_toolkit_astrodynamics-12.0.1.dist-info/top_level.txt,sha256=zOR18699uDYnafgarhL8WU_LmTZY_5NVqutv-flp_x4,5
-open_space_toolkit_astrodynamics-12.0.1.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-open_space_toolkit_astrodynamics-12.0.1.dist-info/RECORD,,
+open_space_toolkit_astrodynamics-12.2.0.dist-info/METADATA,sha256=ucBrKvH3LneOcECxN2_fL-UgEpSRSXsYkC0N4H_vxew,1917
+open_space_toolkit_astrodynamics-12.2.0.dist-info/WHEEL,sha256=EfzGwk6HazeuFg6dmyxdAa9hqpFBtmJ_haGsqhCVSCQ,110
+open_space_toolkit_astrodynamics-12.2.0.dist-info/top_level.txt,sha256=zOR18699uDYnafgarhL8WU_LmTZY_5NVqutv-flp_x4,5
+open_space_toolkit_astrodynamics-12.2.0.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+open_space_toolkit_astrodynamics-12.2.0.dist-info/RECORD,,

ostk/astrodynamics/pytrajectory/pystate.py

@@ -54,13 +54,16 @@ def from_dict(data: dict) -> State:
     - 'vy'/'vy_eci'/'vy_ecef': The y-coordinate of the velocity.
     - 'vz'/'vz_eci'/'vz_ecef': The z-coordinate of the velocity.
     - 'frame': The frame of the state. Required if 'rx', 'ry', 'rz', 'vx', 'vy', 'vz' are provided.
-    - 'q_B_ECI_x': The x-coordinate of the quaternion.
-    - 'q_B_ECI_y': The y-coordinate of the quaternion.
-    - 'q_B_ECI_z': The z-coordinate of the quaternion.
-    - 'q_B_ECI_s': The s-coordinate of the quaternion.
-    - 'w_B_ECI_in_B_x': The x-coordinate of the angular velocity.
-    - 'w_B_ECI_in_B_y': The y-coordinate of the angular velocity.
-    - 'w_B_ECI_in_B_z': The z-coordinate of the angular velocity.
+    - 'q_B_ECI_x': The x-coordinate of the quaternion. Optional.
+    - 'q_B_ECI_y': The y-coordinate of the quaternion. Optional.
+    - 'q_B_ECI_z': The z-coordinate of the quaternion. Optional.
+    - 'q_B_ECI_s': The s-coordinate of the quaternion. Optional.
+    - 'w_B_ECI_in_B_x': The x-coordinate of the angular velocity. Optional.
+    - 'w_B_ECI_in_B_y': The y-coordinate of the angular velocity. Optional.
+    - 'w_B_ECI_in_B_z': The z-coordinate of the angular velocity. Optional.
+    - 'drag_coefficient'/'cd': The drag coefficient. Optional.
+    - 'cross_sectional_area'/'surface_area': The cross-sectional area. Optional.
+    - 'mass': The mass. Optional.
 
 
     Args:
@@ -184,6 +187,27 @@ def from_dict(data: dict) -> State:
             ],
         )
 
+    if "drag_coefficient" in data or "cd" in data:
+        coordinate_subsets.append(CoordinateSubset.drag_coefficient())
+        coordinates = np.append(
+            coordinates,
+            data.get("drag_coefficient", data.get("cd")),
+        )
+
+    if "cross_sectional_area" in data or "surface_area" in data:
+        coordinate_subsets.append(CoordinateSubset.surface_area())
+        coordinates = np.append(
+            coordinates,
+            data.get("cross_sectional_area", data.get("surface_area")),
+        )
+
+    if "mass" in data:
+        coordinate_subsets.append(CoordinateSubset.mass())
+        coordinates = np.append(
+            coordinates,
+            data["mass"],
+        )
+
     return State(
         instant=instant,
         coordinates=coordinates,

ostk/astrodynamics/test/conftest.py

@@ -115,5 +115,5 @@ def aer(azimuth: Angle, elevation: Angle, range: Length) -> LLA:
 
 
 @pytest.fixture
-def trajectory(position: Position) -> Trajectory:
-    return Trajectory.position(position)
+def trajectory(position: Position, instant_1: Instant) -> Trajectory:
+    return Trajectory.position(position.in_frame(Frame.ITRF(), instant_1))

ostk/astrodynamics/test/flight/test_profile.py

@@ -172,7 +172,7 @@ class TestProfile:
         quaternion: Quaternion = Quaternion([0.0, 0.0, 0.0, 1.0], Quaternion.Format.XYZS)
 
         trajectory: Trajectory = Trajectory.position(
-            Position.meters((0.0, 0.0, 0.0), Frame.GCRF())
+            Position.meters((0.0, 0.0, 0.0), Frame.ITRF())
         )
 
         profile: Profile = Profile.inertial_pointing(trajectory, quaternion)

ostk/astrodynamics/test/test_trajectory.py

@@ -1,40 +1,118 @@
 # Apache License 2.0
 
-import ostk.mathematics as mathematics
-
-import ostk.physics as physics
-
-import ostk.astrodynamics as astrodynamics
-
-RealInterval = mathematics.object.RealInterval
-Quaternion = mathematics.geometry.d3.transformation.rotation.Quaternion
-Length = physics.unit.Length
-Angle = physics.unit.Angle
-DateTime = physics.time.DateTime
-Scale = physics.time.Scale
-Duration = physics.time.Duration
-Instant = physics.time.Instant
-Transform = physics.coordinate.Transform
-Frame = physics.coordinate.Frame
-Axes = physics.coordinate.Axes
-DynamicProvider = physics.coordinate.frame.provider.Dynamic
-Environment = physics.Environment
-Earth = physics.environment.object.celestial.Earth
-Trajectory = astrodynamics.Trajectory
-Profile = astrodynamics.flight.Profile
-SatelliteSystem = astrodynamics.flight.system.SatelliteSystem
-Orbit = astrodynamics.trajectory.Orbit
-State = astrodynamics.trajectory.State
-Pass = astrodynamics.trajectory.orbit.Pass
-Kepler = astrodynamics.trajectory.orbit.model.Kepler
-COE = astrodynamics.trajectory.orbit.model.kepler.COE
-SGP4 = astrodynamics.trajectory.orbit.model.sgp4
-Access = astrodynamics.Access
-
-
-def test_trajectory_undefined():
-    trajectory: Trajectory = Trajectory.undefined()
-
-    assert trajectory is not None
-    assert isinstance(trajectory, Trajectory)
-    assert trajectory.is_defined() is False
+import pytest
+
+from ostk.physics.environment.object.celestial import Earth
+from ostk.physics.coordinate.spherical import LLA
+from ostk.physics.coordinate import Position
+from ostk.physics.coordinate import Frame
+from ostk.physics.time import Instant
+from ostk.physics.time import Duration
+from ostk.physics.unit import Derived
+
+from ostk.astrodynamics import Trajectory
+from ostk.astrodynamics.trajectory import State
+
+
+@pytest.fixture
+def start_lla() -> LLA:
+    return LLA.vector([0.0, 0.0, 0.0])
+
+
+@pytest.fixture
+def end_lla() -> LLA:
+    return LLA.vector([1.0, 0.0, 0.0])
+
+
+@pytest.fixture
+def start_instant() -> Instant:
+    return Instant.J2000()
+
+
+@pytest.fixture
+def ground_speed() -> Derived:
+    return Derived(7000.0, Derived.Unit.meter_per_second())
+
+
+@pytest.fixture
+def earth() -> Earth:
+    return Earth.WGS84()
+
+
+@pytest.fixture
+def instants() -> list[Instant]:
+    return [Instant.J2000(), Instant.J2000() + Duration.seconds(10.0)]
+
+
+@pytest.fixture
+def position() -> Position:
+    return Position.meters([0.0, 0.0, 0.0], Frame.ITRF())
+
+
+@pytest.fixture
+def trajectory(position: Position) -> Trajectory:
+    return Trajectory.position(position)
+
+
+@pytest.fixture
+def states(trajectory: Trajectory) -> list[State]:
+    return trajectory.get_states_at(
+        [Instant.J2000(), Instant.J2000() + Duration.seconds(10.0)]
+    )
+
+
+class TestTrajectory:
+    def test_trajectory(self, states: list[State]):
+        assert Trajectory(states) is not None
+
+    def test_is_defined(self, trajectory: Trajectory):
+        assert trajectory.is_defined()
+
+    def test_access_model(self, trajectory: Trajectory):
+        assert trajectory.access_model() is not None
+
+    def test_get_state_at(self, trajectory: Trajectory):
+        assert trajectory.get_state_at(Instant.J2000()) is not None
+
+    def test_get_states_at(self, trajectory: Trajectory):
+        assert (
+            trajectory.get_states_at(
+                [Instant.J2000(), Instant.J2000() + Duration.seconds(10.0)]
+            )
+            is not None
+        )
+
+    def test_trajectory_undefined(self):
+        trajectory: Trajectory = Trajectory.undefined()
+
+        assert trajectory is not None
+        assert isinstance(trajectory, Trajectory)
+        assert trajectory.is_defined() is False
+
+    def test_trajectory_position(self, position: Position):
+        trajectory: Trajectory = Trajectory.position(position)
+
+        assert trajectory is not None
+
+    def test_ground_strip(
+        self,
+        start_lla: LLA,
+        end_lla: LLA,
+        ground_speed: Derived,
+        start_instant: Instant,
+        earth: Earth,
+        instants: list[Instant],
+    ):
+        assert (
+            Trajectory.ground_strip(
+                start_lla, end_lla, ground_speed, start_instant, earth
+            )
+            is not None
+        )
+        assert (
+            Trajectory.ground_strip(start_lla, end_lla, ground_speed, start_instant)
+            is not None
+        )
+
+        assert Trajectory.ground_strip(start_lla, end_lla, instants, earth) is not None
+        assert Trajectory.ground_strip(start_lla, end_lla, instants) is not None

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

@@ -4,7 +4,9 @@ import pytest
 
 from ostk.physics.unit import Length
 from ostk.physics.unit import Angle
+from ostk.physics.time import Instant
 from ostk.physics.environment.gravitational import Earth
+from ostk.physics.environment.object.celestial import Sun
 
 from ostk.astrodynamics.trajectory.orbit.model.kepler import COE
 
@@ -140,6 +142,17 @@ class TestCOE:
         )
         assert COE.compute_radial_distance(7000.0e3, 0.0, 0.0) is not None
 
+        assert COE.compute_ltan(Angle.degrees(270.0), Instant.J2000()) is not None
+        assert (
+            COE.compute_ltan(Angle.degrees(270.0), Instant.J2000(), Sun.default())
+            is not None
+        )
+        assert COE.compute_mean_ltan(Angle.degrees(270.0), Instant.J2000()) is not None
+        assert (
+            COE.compute_mean_ltan(Angle.degrees(270.0), Instant.J2000(), Sun.default())
+            is not None
+        )
+
     def test_from_SI_vector(
         self,
         coe: COE,

ostk/astrodynamics/test/trajectory/test_state.py

@@ -195,6 +195,7 @@ class TestState:
         assert state is not None
         assert isinstance(state, State)
         assert state.get_frame() == Frame.GCRF()
+        assert state.get_size() == 6
 
     def test_from_dict_with_ecef_coordinates(self):
         data = {
@@ -212,6 +213,7 @@ class TestState:
         assert state is not None
         assert isinstance(state, State)
         assert state.get_frame() == Frame.ITRF()
+        assert state.get_size() == 6
 
     def test_from_dict_with_generic_coordinates(self):
         data = {
@@ -230,6 +232,7 @@ class TestState:
         assert state is not None
         assert isinstance(state, State)
         assert state.get_frame() == Frame.GCRF()
+        assert state.get_size() == 6
 
         with pytest.raises(
             ValueError, match="Frame must be provided for generic columns."
@@ -294,6 +297,7 @@ class TestState:
         assert state is not None
         assert isinstance(state, State)
         assert state.get_frame() == Frame.GCRF()
+        assert state.get_size() == 10
 
     def test_from_dict_with_angular_velocity(self):
         data = {
@@ -313,7 +317,95 @@ class TestState:
 
         assert state is not None
         assert isinstance(state, State)
-        assert state.get_frame() == Frame.GCRF()
+        assert state.get_size() == 9
+
+    def test_from_dict_with_drag_coefficient(self):
+        data = {
+            "timestamp": datetime.now(timezone.utc).isoformat(),
+            "rx_eci": 7000.0,
+            "ry_eci": 0.0,
+            "rz_eci": 0.0,
+            "vx_eci": 0.0,
+            "vy_eci": 7.5,
+            "vz_eci": 0.0,
+            "drag_coefficient": 2.2,
+        }
+
+        state: State = State.from_dict(data)
+
+        assert state is not None
+        assert isinstance(state, State)
+        assert state.get_size() == 7
+
+        data = {
+            "timestamp": datetime.now(timezone.utc).isoformat(),
+            "rx_eci": 7000.0,
+            "ry_eci": 0.0,
+            "rz_eci": 0.0,
+            "vx_eci": 0.0,
+            "vy_eci": 7.5,
+            "vz_eci": 0.0,
+            "cd": 2.2,
+        }
+
+        state: State = State.from_dict(data)
+
+        assert state is not None
+        assert isinstance(state, State)
+        assert state.get_size() == 7
+
+    def test_from_dict_with_surface_area(self):
+        data = {
+            "timestamp": datetime.now(timezone.utc).isoformat(),
+            "rx_eci": 7000.0,
+            "ry_eci": 0.0,
+            "rz_eci": 0.0,
+            "vx_eci": 0.0,
+            "vy_eci": 7.5,
+            "vz_eci": 0.0,
+            "surface_area": 2.2,
+        }
+
+        state: State = State.from_dict(data)
+
+        assert state is not None
+        assert isinstance(state, State)
+        assert state.get_size() == 7
+
+        data = {
+            "timestamp": datetime.now(timezone.utc).isoformat(),
+            "rx_eci": 7000.0,
+            "ry_eci": 0.0,
+            "rz_eci": 0.0,
+            "vx_eci": 0.0,
+            "vy_eci": 7.5,
+            "vz_eci": 0.0,
+            "cross_sectional_area": 2.2,
+        }
+
+        state: State = State.from_dict(data)
+
+        assert state is not None
+        assert isinstance(state, State)
+        assert state.get_size() == 7
+
+    def test_from_dict_with_mass(self):
+        data = {
+            "timestamp": datetime.now(timezone.utc).isoformat(),
+            "rx_eci": 7000.0,
+            "ry_eci": 0.0,
+            "rz_eci": 0.0,
+            "vx_eci": 0.0,
+            "vy_eci": 7.5,
+            "vz_eci": 0.0,
+            "mass": 2.2,
+        }
+
+        state: State = State.from_dict(data)
+
+        assert state is not None
+        assert isinstance(state, State)
+        assert state.get_size() == 7
 
     def test_comparators(self, state: State):
         assert (state == state) is True