fix: update imports

tests/test_file_writer_service.py

@@ -2,8 +2,8 @@ from unittest import mock
 
 import numpy as np
 import pytest
-from bec_lib.core import BECMessage, MessageEndpoints, ServiceConfig
-from bec_lib.core.redis_connector import MessageObject
+from bec_lib import MessageEndpoints, ServiceConfig, messages
+from bec_lib.redis_connector import MessageObject
 
 from bec_plugins.services.NIDAQ_writer import NIDAQWriterService
 
@@ -46,7 +46,7 @@ def test_nidaq_scan_status_consumer(nidaq):
 
 
 def test_scan_status_callback(nidaq):
-    scan_status_msg = BECMessage.ScanStatusMessage(scanID="test", status="open", info={})
+    scan_status_msg = messages.ScanStatusMessage(scanID="test", status="open", info={})
     msg_obj = MessageObject(
         topic="test",
         value=scan_status_msg.dumps(),
@@ -85,7 +85,7 @@ def test_nidaq_reads_data_from_strea(nidaq):
 
 @pytest.mark.parametrize("scan_status", ["open", "closed", "aborted", "halted", None])
 def test_nidaq_handle_scan_status(nidaq, scan_status):
-    scan_status_msg = BECMessage.ScanStatusMessage(
+    scan_status_msg = messages.ScanStatusMessage(
         scanID="test", status=scan_status, info={"scan_number": 5}
     )
     nidaq.handle_scan_status(scan_status_msg)
@@ -99,8 +99,8 @@ def test_nidaq_handle_scan_status(nidaq, scan_status):
 
 def test_nidaq_handle_ni_data(nidaq):
     data = [
-        BECMessage.DeviceMessage(signals={"signal1": list(range(10)), "signal2": list(range(10))}),
-        BECMessage.DeviceMessage(
+        messages.DeviceMessage(signals={"signal1": list(range(10)), "signal2": list(range(10))}),
+        messages.DeviceMessage(
             signals={"signal1": list(range(10, 20)), "signal2": list(range(10, 20))}
         ),
     ]