diff --git a/tests/tests_devices/test_mo1_bragg.py b/tests/tests_devices/test_mo1_bragg.py index 89fd6e2..5d48bc2 100644 --- a/tests/tests_devices/test_mo1_bragg.py +++ b/tests/tests_devices/test_mo1_bragg.py @@ -2,27 +2,17 @@ import os import threading import time -from dataclasses import fields from unittest import mock import ophyd import pytest -from bec_lib.messages import ScanQueueMessage, ScanStatusMessage -from bec_server.scan_server.scan_assembler import ScanAssembler -from bec_server.scan_server.scan_queue import RequestBlock from bec_server.scan_server.scan_worker import ScanWorker from bec_server.scan_server.tests.fixtures import scan_server_mock from ophyd.utils import LimitError from ophyd_devices.tests.utils import MockPV # from bec_server.device_server.tests.utils import DMMock -from debye_bec.devices.mo1_bragg.mo1_bragg import ( - Mo1Bragg, - Mo1BraggError, - ScanControlLoadMessage, - ScanControlMode, - ScanControlScanStatus, -) +from debye_bec.devices.mo1_bragg.mo1_bragg import Mo1Bragg, ScanControlMode, ScanControlScanStatus from debye_bec.devices.mo1_bragg.mo1_bragg_devices import MoveType # TODO move this function to ophyd_devices, it is duplicated in csaxs_bec and needed for other pluging repositories