Merge pull request #253 from nyx-space/250-python-spacecraftdynamics-can-only-be-loaded-from-file

Add loads to TrkConfig

Author: ChristopherRabotin

Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "nyx-space"
 build = "build.rs"
-version = "2.0.0-beta-dev"
+version = "2.0.0-beta.0"
 edition = "2021"
 authors = ["Christopher Rabotin <christopher.rabotin@gmail.com>"]
 description = "A high-fidelity space mission toolkit, with orbit propagation, estimation and some systems engineering"

src/python/orbit_determination/trkconfig.rs

@@ -128,6 +128,12 @@ impl TrkConfig {
         Ok(())
     }
 
+    #[classmethod]
+    /// Loads the TrkConfig from its YAML representation
+    fn loads(_cls: &PyType, state: &PyAny) -> Result<Self, ConfigError> {
+        depythonize(state).map_err(|e| ConfigError::InvalidConfig(e.to_string()))
+    }
+
     #[getter]
     fn get_sampling(&self) -> PyResult<Duration> {
         Ok(self.sampling)

tests/python/test_orbit_determination.py

@@ -77,6 +77,8 @@ def test_filter_arc():
     print(trk_cfg_demo.dumps())
     unpkld = pickle.loads(pickle.dumps(trk_cfg_demo))
     assert unpkld == trk_cfg_demo
+    # Check that we can also call the `.loads` function directly
+    assert TrkConfig.loads(trk_cfg_demo.dumps()) == trk_cfg_demo
 
     # Load the trajectory
     traj = TrajectoryLoader(traj_file)