diff --git a/bact_mls_ophyd/devices/pp/quadrupoles.py b/bact_mls_ophyd/devices/pp/quadrupoles.py index cf215ed..5f4b39c 100644 --- a/bact_mls_ophyd/devices/pp/quadrupoles.py +++ b/bact_mls_ophyd/devices/pp/quadrupoles.py @@ -47,9 +47,3 @@ class QuadrupolesCollection(Device, MultiplexerSetMixin): _default_read_attrs = ("sel",) -if __name__ == "__main__": - qc = QuadrupolesCollection(name="qc") - if not qc.connected: - qc.wait_for_connection() - - print(qc.read()) diff --git a/tests/test_quadrupoles.py b/tests/test_quadrupoles.py new file mode 100644 index 0000000..54fdd3b --- /dev/null +++ b/tests/test_quadrupoles.py @@ -0,0 +1,19 @@ +from bact_mls_ophyd.devices.pp.quadrupoles import QuadrupolesCollection +import pytest + + +def create_quad_collection(): + return QuadrupolesCollection(name="qc") + + +def test020_quadrupole_create(): + create_quad_collection() + + +@pytest.mark.skip +def test020_quadrupole_load(): + qc = create_quad_collection() + if not qc.connected: + qc.wait_for_connection() + + print(qc.read())