Fixes

lox-space · Jan 16, 2024 · e8513eb · e8513eb
1 parent 0b8b4d8
commit e8513eb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/crates/lox_py/lox_space.pyi b/crates/lox_py/lox_space.pyi
@@ -13,6 +13,15 @@ class Sun:
 
     def equatorial_radius(self) -> float: ...
 
+class Barycenter:
+    def __init__(self, name: str): ...
+
+    def id(self) -> int: ...
+
+    def name(self) -> str: ...
+
+    def gravitational_parameter(self) -> float: ...
+
 class Planet:
     def __init__(self, name: str): ...
 

diff --git a/crates/lox_py/src/bodies.rs b/crates/lox_py/src/bodies.rs
@@ -84,7 +84,7 @@ impl PyBarycenter {
     }
 
     fn __repr__(&self) -> PyResult<String> {
-        Ok(format!("MinorBody(\"{}\")", self.name()))
+        Ok(format!("Barycenter(\"{}\")", self.name()))
     }
 
     fn __str__(&self) -> PyResult<String> {
@@ -130,7 +130,7 @@ impl PyPlanet {
     }
 
     fn __repr__(&self) -> PyResult<String> {
-        Ok(format!("MinorBody(\"{}\")", self.name()))
+        Ok(format!("Planet(\"{}\")", self.name()))
     }
 
     fn __str__(&self) -> PyResult<String> {
@@ -219,7 +219,7 @@ impl PySatellite {
     }
 
     fn __repr__(&self) -> PyResult<String> {
-        Ok(format!("MinorBody(\"{}\")", self.name()))
+        Ok(format!("Satellite(\"{}\")", self.name()))
     }
 
     fn __str__(&self) -> PyResult<String> {