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:
@ -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
|
||||
|
@ -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())
|
||||
#
|
||||
|
Reference in New Issue
Block a user