bec_widgets/utils/bec_dispatcher.py

@@ -16,7 +16,7 @@ from qtpy.QtCore import Signal as pyqtSignal
 
 logger = bec_logger.logger
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from bec_lib.endpoints import EndpointInfo
 
     from bec_widgets.utils.rpc_server import RPCServer

bec_widgets/utils/colors.py

@@ -11,7 +11,7 @@ from pydantic_core import PydanticCustomError
 from qtpy.QtGui import QColor
 from qtpy.QtWidgets import QApplication
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from bec_qthemes._main import AccentColors
 
 

bec_widgets/utils/rpc_server.py

@@ -18,7 +18,7 @@ from bec_widgets.utils import BECDispatcher
 from bec_widgets.utils.bec_connector import BECConnector
 from bec_widgets.utils.error_popups import ErrorPopupUtility
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from bec_lib import messages
     from qtpy.QtCore import QObject
 else:

bec_widgets/widgets/containers/dock/dock.py

@@ -15,7 +15,7 @@ from bec_widgets.utils.container_utils import WidgetContainerUtils
 
 logger = bec_logger.logger
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from qtpy.QtWidgets import QWidget
 
     from bec_widgets.widgets.containers.dock.dock_area import BECDockArea

bec_widgets/widgets/editors/scan_metadata/_metadata_widgets.py

@@ -33,7 +33,7 @@ from bec_widgets.widgets.editors.scan_metadata._util import (
     field_precision,
 )
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from pydantic.fields import FieldInfo
 
 logger = bec_logger.logger

bec_widgets/widgets/editors/scan_metadata/_util.py

@@ -9,7 +9,7 @@ from annotated_types import Ge, Gt, Le, Lt
 from bec_lib.logger import bec_logger
 from pydantic_core import PydanticUndefined
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from pydantic.fields import FieldInfo
 
 logger = bec_logger.logger

bec_widgets/widgets/editors/scan_metadata/scan_metadata.py

@@ -29,7 +29,7 @@ from bec_widgets.widgets.editors.scan_metadata.additional_metadata_table import
     AdditionalMetadataTable,
 )
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from pydantic.fields import FieldInfo
 
 logger = bec_logger.logger

bec_widgets/widgets/services/bec_status_box/bec_status_box.py

@@ -16,7 +16,7 @@ from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.utils.compact_popup import CompactPopupWidget
 from bec_widgets.widgets.services.bec_status_box.status_item import StatusItem
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from bec_lib.client import BECClient
 
 # TODO : Put normal imports back when Pydantic gets faster

bec_widgets/widgets/services/device_browser/device_item/device_item.py

@@ -7,7 +7,7 @@ from qtpy.QtCore import QMimeData, Qt
 from qtpy.QtGui import QDrag
 from qtpy.QtWidgets import QApplication, QHBoxLayout, QLabel, QWidget
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from qtpy.QtGui import QMouseEvent
 
 logger = bec_logger.logger

bec_widgets/widgets/utility/logpanel/logpanel.py

@@ -51,7 +51,7 @@ from bec_widgets.widgets.utility.logpanel._util import (
     simple_color_format,
 )
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from PySide6.QtCore import SignalInstance
 
 logger = bec_logger.logger

tests/end-2-end/test_plotting_framework_e2e.py

@@ -139,7 +139,9 @@ def test_async_plotting(qtbot, bec_client_lib, connected_client_gui_obj):
         return client.history[-1].metadata.bec["scan_id"] == status.scan.scan_id
 
     qtbot.waitUntil(_wait_for_scan_in_history, timeout=10000)
-    last_scan_data = client.history[-1]
+
+    item = queue.scan_storage.storage[-1]
+    last_scan_data = item.data
     # check plotted data
     x_data, y_data = curve.get_data()
     assert np.array_equal(x_data, np.linspace(0, len(y_data) - 1, len(y_data)))
@@ -151,8 +153,6 @@ def test_async_plotting(qtbot, bec_client_lib, connected_client_gui_obj):
     x_data_display, y_data_display = curve._get_displayed_data()
     # Should be not more than 1% difference, actually be closer but this might be flaky
     assert np.isclose(x_data_display[-1], x_data[-1], rtol=0.01)
-    # Downsampled data should be smaller than original data
-    assert len(y_data_display) < len(y_data)
 
 
 def test_rpc_image(qtbot, bec_client_lib, connected_client_gui_obj):

tests/end-2-end/test_rpc_widgets_e2e.py

@@ -136,6 +136,8 @@ def test_all_available_widgets(
         # Skip private attributes
         if widget_name.startswith("_"):
             continue
+        if widget_name == "VSCodeEditor":
+            continue  # Not installed in docker image for CI, so we skip it.
         dock_area, dock, widget = create_widget(
             qtbot, gui, getattr(gui.available_widgets, widget_name)
         )
@@ -653,23 +655,6 @@ def test_widgets_e2e_text_box(
     maybe_remove_dock_area(qtbot, gui, dock_area, random_generator_from_seed)
 
 
-@pytest.mark.timeout(PYTEST_TIMEOUT)
-def test_widgets_e2e_vs_code_editor(
-    qtbot, connected_gui_and_bec_with_scope_session, random_generator_from_seed
-):
-    """Test the VSCodeEditor widget"""
-    gui = connected_gui_and_bec_with_scope_session
-    bec = gui._client
-    # Create dock_area, dock, widget
-    dock_area, dock, widget = create_widget(qtbot, gui, gui.available_widgets.VSCodeEditor)
-
-    # No rpc calls to check so far
-
-    # Test removing the widget, or leaving it open for the next test
-    maybe_remove_widget(qtbot, gui, dock, widget, random_generator_from_seed)
-    maybe_remove_dock_area(qtbot, gui, dock_area, random_generator_from_seed)
-
-
 @pytest.mark.timeout(PYTEST_TIMEOUT)
 def test_widgets_e2e_waveform(
     qtbot, connected_gui_and_bec_with_scope_session, random_generator_from_seed

tests/unit_tests/test_device_browser.py

@@ -8,7 +8,7 @@ from bec_widgets.widgets.services.device_browser.device_browser import DeviceBro
 
 from .client_mocks import mocked_client
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from qtpy.QtWidgets import QListWidgetItem
 
     from bec_widgets.widgets.services.device_browser import DeviceItem