Add unit test set_dof.py

tests/test_maintel_set_dof.py

@@ -0,0 +1,84 @@
+# This file is part of ts_standardscripts
+#
+# Developed for the LSST Telescope and Site Systems.
+# This product includes software developed by the LSST Project
+# (https://www.lsst.org).
+# See the COPYRIGHT file at the top-level directory of this distribution
+# for details of code ownership.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import random
+import types
+import unittest
+
+import numpy as np
+from lsst.ts import standardscripts
+from lsst.ts.observatory.control.maintel.mtcs import MTCS, MTCSUsages
+from lsst.ts.observatory.control.utils.enums import DOFName
+from lsst.ts.standardscripts.maintel import SetDOF
+
+random.seed(47)  # for set_random_lsst_dds_partition_prefix
+
+
+class TestSetDOF(standardscripts.BaseScriptTestCase, unittest.IsolatedAsyncioTestCase):
+    async def basic_make_script(self, index):
+        self.script = SetDOF(index=index)
+
+        # Mock the MTCS
+        self.script.mtcs = MTCS(
+            domain=self.script.domain,
+            intended_usage=MTCSUsages.DryTest,
+            log=self.script.log,
+        )
+        self.script.mtcs.rem.mtaos = unittest.mock.AsyncMock()
+        self.script.mtcs.rem.mtaos.cmd_offsetDOF.attach_mock(
+            unittest.mock.Mock(
+                return_value=types.SimpleNamespace(value=np.zeros(len(DOFName)))
+            ),
+            "DataType",
+        )
+        self.script.mtcs.rem.mtaos.configure_mock(
+            **{
+                "evt_degreeOfFreedom.aget.return_value": types.SimpleNamespace(
+                    aggregatedDoF=np.zeros(len(DOFName))
+                ),
+            }
+        )
+
+        self.script.mtcs.assert_all_enabled = unittest.mock.AsyncMock()
+
+        return (self.script,)
+
+    async def test_run(self) -> None:
+        # Start the test itself
+        async with self.make_script():
+            config_dofs = {"M2_dz": 0.2, "Cam_dy": 0.3, "M1M3_B1": 0.5, "M2_B14": 0.7}
+
+            await self.configure_script(**config_dofs)
+
+            # Run the script
+            await self.run_script()
+
+            self.script.mtcs.rem.mtaos.cmd_offsetDOF.DataType.assert_called()
+            self.script.mtcs.rem.mtaos.cmd_offsetDOF.start.assert_awaited_once()
+
+    async def test_executable(self) -> None:
+        scripts_dir = standardscripts.get_scripts_dir()
+        script_path = scripts_dir / "maintel" / "set_dof.py"
+        await self.check_executable(script_path)
+
+
+if __name__ == "__main__":
+    unittest.main()