Add assert_slew_planet method to test_maintel_track_target.py

lsst-ts · Nov 28, 2023 · fdab7db · fdab7db
1 parent f629709
commit fdab7db
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_maintel_track_target.py b/tests/test_maintel_track_target.py
@@ -355,6 +355,16 @@ def assert_slew_fails(self):
         self.script.tcs.slew_object.assert_not_awaited()
         self.script.tcs.stop_tracking.assert_awaited_once()
 
+    def assert_slew_planet(self, planet_name):
+        planet_enum = Planets[planet_name]
+        self.script.tcs.slew_to_planet.assert_awaited_once()
+        self.script.tcs.slew_to_planet.assert_awaited_with(
+            planet=planet_enum,
+            rot_sky=self.script.config.rot_value,
+            slew_timeout=self.script.config.slew_timeout,
+        )
+        self.script.tcs.stop_tracking.assert_not_awaited()
+
 
 if __name__ == "__main__":
     unittest.main()