0
0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 03:31:50 +02:00

refactor: fix bec_lib imports

This commit is contained in:
2023-11-13 08:29:07 +01:00
parent b38e942acb
commit 5ec2b08e34
13 changed files with 31 additions and 35 deletions

View File

@ -1,8 +1,8 @@
from unittest.mock import Mock
import pytest
from bec_lib.core.BECMessage import ScanMessage
from bec_lib.core.connector import MessageObject
from bec_lib.messages import ScanMessage
from bec_lib.connector import MessageObject
# TODO: find a better way to mock singletons
from bec_widgets.bec_dispatcher import _BECDispatcher

View File

@ -2,11 +2,10 @@ from unittest import mock
import numpy as np
import pytest
from bec_lib.core import BECMessage
from bec_lib import messages, RedisConnector
from pytestqt import qtbot
import threading
from bec_lib.core import RedisConnector
from bec_widgets.examples.stream_plot.stream_plot import StreamPlot
@ -144,7 +143,7 @@ def test_on_dap_update(qtbot, stream_app):
# }
#
# # Assume the RedisConnector client would return this data when new_proj is called
# mock_message = mock.MagicMock(spec=BECMessage.DeviceMessage)
# mock_message = mock.MagicMock(spec=messages.DeviceMessage)
# mock_message.__getitem__.side_effect = lambda key: mock_data[key]
# stream_app.client.producer.get = mock.MagicMock(return_value=mock_message.dumps())
#