export from internal gitlab

tests/test_galil.py

@@ -0,0 +1,60 @@
+import pytest
+from ophyd_devices.galil.galil_ophyd import GalilMotor
+
+from utils import SocketMock
+
+
+@pytest.mark.parametrize(
+    "pos,msg,sign",
+    [
+        (1, b" -12800\n\r", 1),
+        (-1, b" -12800\n\r", -1),
+    ],
+)
+def test_axis_get(pos, msg, sign):
+    leyey = GalilMotor(
+        "H",
+        name="leyey",
+        host="mpc2680.psi.ch",
+        port=8081,
+        sign=sign,
+        socket_cls=SocketMock,
+    )
+    leyey.controller.sock.flush_buffer()
+    leyey.stage()
+    leyey.controller.sock.buffer_recv = msg
+    val = leyey.read()
+    assert val["leyey"]["value"] == pos
+    assert leyey.readback.get() == pos
+
+
+@pytest.mark.parametrize(
+    "target_pos,socket_put_messages,socket_get_messages",
+    [
+        (
+            0,
+            [
+                b"MG allaxref\r",
+                b"MG_XQ0\r",
+                b"naxis=7\r",
+                b"ntarget=0.000\r",
+                b"movereq=1\r",
+                b"XQ#NEWPAR\r",
+            ],
+            [
+                b"1.00",
+                b"-1",
+                b":",
+                b":",
+                b":",
+                b":",
+            ],
+        ),
+    ],
+)
+def test_axis_put(target_pos, socket_put_messages, socket_get_messages):
+    leyey = GalilMotor("H", name="leyey", host="mpc2680.psi.ch", port=8081, socket_cls=SocketMock)
+    leyey.controller.sock.flush_buffer()
+    leyey.controller.sock.buffer_recv = socket_get_messages
+    leyey.user_setpoint.put(target_pos)
+    assert leyey.controller.sock.buffer_put == socket_put_messages