bec-widgets 0.83.1__py3-none-any.whl → 0.85.0__py3-none-any.whl

bec_widgets/widgets/figure/plots/waveform/waveform_curve.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any, Literal, Optional
 
+import numpy as np
 import pyqtgraph as pg
 from pydantic import BaseModel, Field, field_validator
 from pydantic_core import PydanticCustomError
@@ -28,7 +29,7 @@ class Signal(BaseModel):
     """The configuration of a signal in the 1D waveform widget."""
 
     source: str
-    x: SignalData  # TODO maybe add metadata for config gui later
+    x: Optional[SignalData] = None
     y: SignalData
     z: Optional[SignalData] = None
     dap: Optional[str] = None
@@ -248,9 +249,15 @@ class BECCurve(BECConnector, pg.PlotDataItem):
         Returns:
             tuple[np.ndarray,np.ndarray]: X and Y data of the curve.
         """
-        x_data, y_data = self.getData()
+        try:
+            x_data, y_data = self.getData()
+        except TypeError:
+            x_data, y_data = np.array([]), np.array([])
         return x_data, y_data
 
+    def clear_data(self):
+        self.setData([], [])
+
     def remove(self):
         """Remove the curve from the plot."""
         # self.parent_item.removeItem(self)

{bec_widgets-0.83.1.dist-info → bec_widgets-0.85.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bec_widgets
-Version: 0.83.1
+Version: 0.85.0
 Summary: BEC Widgets
 Project-URL: Bug Tracker, https://gitlab.psi.ch/bec/bec_widgets/issues
 Project-URL: Homepage, https://gitlab.psi.ch/bec/bec_widgets

{bec_widgets-0.83.1.dist-info → bec_widgets-0.85.0.dist-info}/RECORD

@@ -2,11 +2,11 @@
 .gitlab-ci.yml,sha256=zvb4A6QI5lQTsdfI5nPPL-tUNfcrz__SQjxW03QZ5Ek,8204
 .pylintrc,sha256=eeY8YwSI74oFfq6IYIbCqnx3Vk8ZncKaatv96n_Y8Rs,18544
 .readthedocs.yaml,sha256=aSOc277LqXcsTI6lgvm_JY80lMlr69GbPKgivua2cS0,603
-CHANGELOG.md,sha256=XIRZNOiAJ6U-HWIqQIZQrJ-seB-mrY8_bRoAA-QMH6w,6629
+CHANGELOG.md,sha256=qfCubB8BgPi9QiPtGGeDxjPct_eSvK1Ep9UIP9Rgh34,7430
 LICENSE,sha256=YRKe85CBRyP7UpEAWwU8_qSIyuy5-l_9C-HKg5Qm8MQ,1511
-PKG-INFO,sha256=DQQ4PF3OQf7ROCg2O2cIkgZoV2jsXd_1z6I-P2Z5Px0,1308
+PKG-INFO,sha256=WWcaoP_OMmYA22TW83DKHT9_qQS5IpZxIOV5wcyUPkM,1308
 README.md,sha256=Od69x-RS85Hph0-WwWACwal4yUd67XkEn4APEfHhHFw,2649
-pyproject.toml,sha256=LyalVcXB0KNBOfUgjPnaHLv6kQFZCn7vvbFlnjZ_bnc,2357
+pyproject.toml,sha256=bCu_XxdOirPXbnbNVdYtCKmbhJCJFxnxINDiZU02bnE,2357
 .git_hooks/pre-commit,sha256=n3RofIZHJl8zfJJIUomcMyYGFi_rwq4CC19z0snz3FI,286
 .gitlab/issue_templates/bug_report_template.md,sha256=gAuyEwl7XlnebBrkiJ9AqffSNOywmr8vygUFWKTuQeI,386
 .gitlab/issue_templates/documentation_update_template.md,sha256=FHLdb3TS_D9aL4CYZCjyXSulbaW5mrN2CmwTaeLPbNw,860
@@ -17,7 +17,7 @@ bec_widgets/assets/bec_widgets_icon.png,sha256=K8dgGwIjalDh9PRHUsSQBqgdX7a00nM3i
 bec_widgets/assets/terminal_icon.png,sha256=bJl7Tft4Fi2uxvuXI8o14uMHnI9eAWKSU2uftXCH9ws,3889
 bec_widgets/cli/__init__.py,sha256=d0Q6Fn44e7wFfLabDOBxpcJ1DPKWlFunGYDUBmO-4hA,22
 bec_widgets/cli/auto_updates.py,sha256=DyBV3HnjMSH-cvVkYNcDiYKVf0Xut4Qy2qGQqkW47Bw,4833
-bec_widgets/cli/client.py,sha256=MMDy1edr1j9klgVGZZmD1dzLxi74gp75Jf4FNed9al8,60084
+bec_widgets/cli/client.py,sha256=4zUI7gm0J2gdKCvfDuK-47PRZ2x3WRchObhhmt0FMsU,61843
 bec_widgets/cli/client_utils.py,sha256=cDhabblwaP88a0jlVpbn_RWWKVbsyjhmmGtMh9gesEw,12388
 bec_widgets/cli/generate_cli.py,sha256=Ea5px9KblUlcGg-1JbJBTIU7laGg2n8PM7Efw9WVVzM,5889
 bec_widgets/cli/rpc_register.py,sha256=QxXUZu5XNg00Yf5O3UHWOXg3-f_pzKjjoZYMOa-MOJc,2216
@@ -25,8 +25,7 @@ bec_widgets/cli/rpc_wigdet_handler.py,sha256=6kQng2DyS6rhLJqSJ7xa0kdgSxp-35A2upc
 bec_widgets/cli/server.py,sha256=FZkqsjUHIkCUFMKUFm7ls_eslXvhIFzLpINEYxeWP5s,7722
 bec_widgets/examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bec_widgets/examples/jupyter_console/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bec_widgets/examples/jupyter_console/jupyter_console_window.py,sha256=S6OTEp8i1sWRyfB9ZBhl64ZaqBvjonI9xqqtWv2Amp4,5540
-bec_widgets/examples/jupyter_console/jupyter_console_window.ui,sha256=2A2mNTUMZBYygz8K4qWzrcjnNqZBMVyeHm26iLZVRWI,1473
+bec_widgets/examples/jupyter_console/jupyter_console_window.py,sha256=iBJD3bNE2tH2sxm2gsdzbdHVEyQRGMMTCzh7utG6lPg,6638
 bec_widgets/examples/plugin_example_pyside/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bec_widgets/examples/plugin_example_pyside/main.py,sha256=xdC6RWSRt1rW8Prj0CsDeCPct6-_j3__oJmdgogB5PI,505
 bec_widgets/examples/plugin_example_pyside/registertictactoe.py,sha256=VNFkHc5Sc30lRKzOFJbhArCHGkp_wRxOeJjZbmaAHRU,448
@@ -40,7 +39,7 @@ bec_widgets/qt_utils/toolbar.py,sha256=UUvC_AwUW3d6MqAzRJIBBqNWLrvBQI9UvekJ9Dw1A
 bec_widgets/utils/__init__.py,sha256=1930ji1Jj6dVuY81Wd2kYBhHYNV-2R0bN_L4o9zBj1U,533
 bec_widgets/utils/bec_connector.py,sha256=NypWbIrqb2ls3SIpflM6KihidV9fkroiJu2tQk6KwOA,9604
 bec_widgets/utils/bec_designer.py,sha256=ak3G8FdojUPjVBBwdPXw7tN5P2Uxr-SSoQt394jXeAA,4308
-bec_widgets/utils/bec_dispatcher.py,sha256=dghOw63Ql6rQ-zH1ZaYU6kT9Dg5kNDcZwEdu2YPBJeU,6150
+bec_widgets/utils/bec_dispatcher.py,sha256=fhI7_X0kSZCtXyR55Qn-w7BfNdk2Roc1Tyx0bx3bjoE,6195
 bec_widgets/utils/bec_table.py,sha256=nA2b8ukSeUfquFMAxGrUVOqdrzMoDYD6O_4EYbOG2zk,717
 bec_widgets/utils/bec_widget.py,sha256=eC7jhCqyABDf_wGbM6cCdtSF4__rpnVs2Ir1XzGHrCs,238
 bec_widgets/utils/colors.py,sha256=RAGaA4jdvsFSxGlbhSBYL5mdlwS-9rq45tJM7U-7xXs,10587
@@ -79,6 +78,12 @@ bec_widgets/widgets/color_button/color_button.pyproject,sha256=LUYF4VkDOB4ttVe7Y
 bec_widgets/widgets/color_button/color_button_plugin.py,sha256=CvO3L6r8pBd-0DNuqjQ7uKF9u1mU6xFs23sS_dsj4tw,1256
 bec_widgets/widgets/color_button/register_color_button.py,sha256=ZAx3t7G2VI2S-1qcEof31Xi9O0X8deLeZNgS651L1JI,475
 bec_widgets/widgets/color_button/assets/color_button.png,sha256=dkjgGHSdxu98nM68O16fGpxBjxkkeLN8iu9M_8i5-NY,5618
+bec_widgets/widgets/color_map_selector/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bec_widgets/widgets/color_map_selector/color_map_selector.py,sha256=V0nmY2_X76xJmLXOfjW6rG7ipJKFT8swo3bkJZDXBGE,3751
+bec_widgets/widgets/color_map_selector/color_map_selector.pyproject,sha256=sGQE4s6gia1PIz9p6c2mZX12geCk8G384gjL2PE_qEc,36
+bec_widgets/widgets/color_map_selector/color_map_selector_plugin.py,sha256=9V7xvq2SB_YZU7W1E6aEKx38REOTspFNc7x57Fyd9mY,1436
+bec_widgets/widgets/color_map_selector/register_color_map_selector.py,sha256=crZsJcH4z1DHfsfA7iHiQFDggslNyzfuHfo-YdAoca4,514
+bec_widgets/widgets/color_map_selector/assets/color_map_selector_icon.png,sha256=o8zRmMvXhGhMyBklH_Vfjn-jpDL4Nw03fLqDk_wx_JI,11975
 bec_widgets/widgets/console/console.py,sha256=IW1OerycmS-cm8CKGFig44Qye8mxsmVcKvLHAc3lXco,17845
 bec_widgets/widgets/device_box/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bec_widgets/widgets/device_box/device_box.py,sha256=ofYQJLY02pitmAJG0Y3GWBg8txNmaJDWhQVe9KFQDYA,6938
@@ -102,7 +107,7 @@ bec_widgets/widgets/dock/__init__.py,sha256=B7foHt02gnhM7mFksa7GJVwT7n0j_JvYDCt6
 bec_widgets/widgets/dock/dock.py,sha256=joymi8NRoIuzuugUj9ccF9e1m57HwLQhhMmjaWiwTnM,7597
 bec_widgets/widgets/dock/dock_area.py,sha256=WKIt61v7w2YXahfIL4nddWHPfpTpw52uphX4QCbS3q0,7913
 bec_widgets/widgets/figure/__init__.py,sha256=3hGx_KOV7QHCYAV06aNuUgKq4QIYCjUTad-DrwkUaBM,44
-bec_widgets/widgets/figure/figure.py,sha256=5N9GvUMwOOcQgIR78olVhXjIb0W64vIpG6_8WPcoNsc,32014
+bec_widgets/widgets/figure/figure.py,sha256=vuaeAT6XZaK37cLQh_KXX0pE-z6dLkmc_QNe4eydQpk,28520
 bec_widgets/widgets/figure/plots/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bec_widgets/widgets/figure/plots/axis_settings.py,sha256=CCvJwo1RjpnLJG9TmGqf9fyPpUNyLT8k7pPEnqVuaGM,2281
 bec_widgets/widgets/figure/plots/axis_settings.ui,sha256=zMKZK6lH_3KJGO4RA_paXAH7UzZJ4Snlil3MK4pK3L8,11589
@@ -114,8 +119,8 @@ bec_widgets/widgets/figure/plots/image/image_processor.py,sha256=GeTtWjbldy6VejM
 bec_widgets/widgets/figure/plots/motor_map/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bec_widgets/widgets/figure/plots/motor_map/motor_map.py,sha256=FH3ZSYThGco98jS29r9EGcIh5fYx8e5eOs_nYJNrr48,18210
 bec_widgets/widgets/figure/plots/waveform/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bec_widgets/widgets/figure/plots/waveform/waveform.py,sha256=e401GGvUF11rZKf4onB4zYlcUWIv8Zsnm4GcysaXi6U,29226
-bec_widgets/widgets/figure/plots/waveform/waveform_curve.py,sha256=yQZGPKs--3X_9Qg2pv0GUiL5WLBQVC3z_PJKRnsHqPU,8293
+bec_widgets/widgets/figure/plots/waveform/waveform.py,sha256=fH3gl1KFVUiKUhfk4UNnBzhW-gyeZq0K_xgrdeKq8PA,52537
+bec_widgets/widgets/figure/plots/waveform/waveform_curve.py,sha256=rQow0EkMtoETJ4xNRRtHbpPuJ4AmNjtn9fx4IM5enh4,8436
 bec_widgets/widgets/jupyter_console/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bec_widgets/widgets/jupyter_console/jupyter_console.py,sha256=ioLYJL31RdBoAOGFSS8PVSnUhkWPWmLC3tiKp7CouO8,2251
 bec_widgets/widgets/motor_map/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -223,7 +228,7 @@ tests/references/SpinnerWidget/SpinnerWidget_linux.png,sha256=OyiGxyQx0XCKEa1OeA
 tests/references/SpinnerWidget/SpinnerWidget_started_darwin.png,sha256=NA7dOdKY-leFv8JI_5x3OIIY-XlSXSTIflVquz0UUZc,13784
 tests/references/SpinnerWidget/SpinnerWidget_started_linux.png,sha256=NA7dOdKY-leFv8JI_5x3OIIY-XlSXSTIflVquz0UUZc,13784
 tests/unit_tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit_tests/client_mocks.py,sha256=_gtf12kTxmwaXKwSOG5Cc0dqVEY1PHi-hmqdTdLhLTk,4752
+tests/unit_tests/client_mocks.py,sha256=nyI1Qi5ZcDtdfYdNjf46rrL5d5vkVxXcgG0hIZRolO8,5051
 tests/unit_tests/conftest.py,sha256=KrnktXPWmZhnKNue-xGWOLD1XGEvdz9Vf7V2eO3XQ3A,596
 tests/unit_tests/test_bec_connector.py,sha256=zGDfNHwLFZTbpyX6-yc7Pwzr2jWO_HGZ8T4NFCNo4IE,2444
 tests/unit_tests/test_bec_dispatcher.py,sha256=rYPiRizHaswhGZw55IBMneDFxmPiCCLAZQBqjEkpdyY,3992
@@ -234,11 +239,12 @@ tests/unit_tests/test_bec_motor_map.py,sha256=dSYopbZS8lGD9cB26Kwmqw5zBoKCZs8t7D
 tests/unit_tests/test_bec_queue.py,sha256=u-uc-iZeGAS8P90o6Cxy5oz_60zHpirGAu04OgQPDXw,4598
 tests/unit_tests/test_bec_status_box.py,sha256=gZdjyy9DNuUP9UwleTLj2Dp5HUImiqnkHjXWiqL0Q-o,4868
 tests/unit_tests/test_client_utils.py,sha256=CBdWIVJ_UiyFzTJnX3XJm4PGw2uXhFvRCP_Y9ifckbw,2630
+tests/unit_tests/test_color_map_selector.py,sha256=NVnKrz5kFUKgljm7HslIpMII-IpzPAKHIX_KeqpNUew,1518
 tests/unit_tests/test_color_validation.py,sha256=xbFbtFDia36XLgaNrX2IwvAX3IDC_Odpj5BGoJSgiIE,2389
 tests/unit_tests/test_crosshair.py,sha256=3OMAJ2ZaISYXMOtkXf1rPdy94vCr8njeLi6uHblBL9Q,5045
 tests/unit_tests/test_device_box.py,sha256=q9IVFpt1NF3TBF0Jhk-I-LRiuvvHG3FGUalw4jEYwVo,3431
 tests/unit_tests/test_device_input_base.py,sha256=r1tI7BFAhpv7V7gf_n5gjusyrBFOfuCqIkdVg7YA7vY,2444
-tests/unit_tests/test_device_input_widgets.py,sha256=Q40xNKGvJT2dvNttRH68WZi0au8PwpUgBi2EB4LXfC8,5841
+tests/unit_tests/test_device_input_widgets.py,sha256=39MtgF-Q67UWz6qapyYP4ukDEUOD81iEJ_jhATyG7dM,5889
 tests/unit_tests/test_generate_cli_client.py,sha256=ng-eV5iF7Dhm-6YpxYo99CMY0KgqoaZBQNkMeKULDBU,3355
 tests/unit_tests/test_generate_plugin.py,sha256=9603ucZChM-pYpHadzsR94U1Zec1KZT34WedX9qzgMo,4464
 tests/unit_tests/test_motor_map_widget.py,sha256=3nbINg3NYvWUrrGGMRPs8SDtePjXhoehSY_CShFGvEI,7507
@@ -256,7 +262,7 @@ tests/unit_tests/test_stop_button.py,sha256=tpQanzBUyl7qLXjbMUQqm3U3vShbKKARcnLp
 tests/unit_tests/test_text_box_widget.py,sha256=cT0uEHt_6d-FwST0A_wE9sFW9E3F_nJbKhuBAeU4yHg,1862
 tests/unit_tests/test_toggle.py,sha256=Amzgres7te0tTQIDR2WMKSx9Kce44TxMpIPR6HZygXQ,832
 tests/unit_tests/test_vscode_widget.py,sha256=G1G7nCQGXFUn0BnMECE7mHmAm0C6pYx1JpEi_XEhodY,2682
-tests/unit_tests/test_waveform1d.py,sha256=inc-CyRA_3-PE99WnR6vDdfRE1vgp-ZoLVJ-Qbq0qFY,15967
+tests/unit_tests/test_waveform1d.py,sha256=ZuHCvGubMuaLIzaMWDvmBUhgzUHCDLdvTZqIOfBKaZg,22713
 tests/unit_tests/test_website_widget.py,sha256=fBADIJJBAHU4Ro7u95kdemFVNv196UOcuO9oLHuHt8A,761
 tests/unit_tests/test_widget_io.py,sha256=FeL3ZYSBQnRt6jxj8VGYw1cmcicRQyHKleahw7XIyR0,3475
 tests/unit_tests/test_yaml_dialog.py,sha256=SEvUgC_poWC6fAoHVWolaORpgMFc7c0Xqqk9cFvHSvo,5826
@@ -265,8 +271,8 @@ tests/unit_tests/test_configs/config_device_no_entry.yaml,sha256=hdvue9KLc_kfNzG
 tests/unit_tests/test_configs/config_scan.yaml,sha256=vo484BbWOjA_e-h6bTjSV9k7QaQHrlAvx-z8wtY-P4E,1915
 tests/unit_tests/test_msgs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/unit_tests/test_msgs/available_scans_message.py,sha256=m_z97hIrjHXXMa2Ex-UvsPmTxOYXfjxyJaGkIY6StTY,46532
-bec_widgets-0.83.1.dist-info/METADATA,sha256=DQQ4PF3OQf7ROCg2O2cIkgZoV2jsXd_1z6I-P2Z5Px0,1308
-bec_widgets-0.83.1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-bec_widgets-0.83.1.dist-info/entry_points.txt,sha256=3otEkCdDB9LZJuBLzG4pFLK5Di0CVybN_12IsZrQ-58,166
-bec_widgets-0.83.1.dist-info/licenses/LICENSE,sha256=YRKe85CBRyP7UpEAWwU8_qSIyuy5-l_9C-HKg5Qm8MQ,1511
-bec_widgets-0.83.1.dist-info/RECORD,,
+bec_widgets-0.85.0.dist-info/METADATA,sha256=WWcaoP_OMmYA22TW83DKHT9_qQS5IpZxIOV5wcyUPkM,1308
+bec_widgets-0.85.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+bec_widgets-0.85.0.dist-info/entry_points.txt,sha256=3otEkCdDB9LZJuBLzG4pFLK5Di0CVybN_12IsZrQ-58,166
+bec_widgets-0.85.0.dist-info/licenses/LICENSE,sha256=YRKe85CBRyP7UpEAWwU8_qSIyuy5-l_9C-HKg5Qm8MQ,1511
+bec_widgets-0.85.0.dist-info/RECORD,,

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "bec_widgets"
-version = "0.83.1"
+version = "0.85.0"
 description = "BEC Widgets"
 requires-python = ">=3.10"
 classifiers = [

tests/unit_tests/client_mocks.py

@@ -4,7 +4,7 @@ from unittest.mock import MagicMock, patch
 import fakeredis
 import pytest
 from bec_lib.client import BECClient
-from bec_lib.device import Positioner
+from bec_lib.device import Positioner, ReadoutPriority
 from bec_lib.devicemanager import DeviceContainer
 from bec_lib.redis_connector import RedisConnector
 
@@ -12,11 +12,12 @@ from bec_lib.redis_connector import RedisConnector
 class FakeDevice:
     """Fake minimal positioner class for testing."""
 
-    def __init__(self, name, enabled=True):
+    def __init__(self, name, enabled=True, readout_priority=ReadoutPriority.MONITORED):
         self.name = name
         self.enabled = enabled
         self.signals = {self.name: {"value": 1.0}}
         self.description = {self.name: {"source": self.name, "dtype": "number", "shape": []}}
+        self.readout_priority = readout_priority
 
     def __contains__(self, item):
         return item == self.name
@@ -43,8 +44,15 @@ class FakeDevice:
 
 
 class FakePositioner(FakeDevice):
-    def __init__(self, name, enabled=True, limits=None, read_value=1.0):
-        super().__init__(name, enabled)
+    def __init__(
+        self,
+        name,
+        enabled=True,
+        limits=None,
+        read_value=1.0,
+        readout_priority=ReadoutPriority.MONITORED,
+    ):
+        super().__init__(name, enabled, readout_priority)
         self.limits = limits if limits is not None else [0, 0]
         self.read_value = read_value
         self.name = name
@@ -110,6 +118,7 @@ DEVICES = [
     FakeDevice("bpm3a"),
     FakeDevice("bpm3i"),
     FakeDevice("eiger"),
+    FakeDevice("async_device", readout_priority=ReadoutPriority.ASYNC),
     Positioner("test", limits=[-10, 10], read_value=2.0),
 ]
 

tests/unit_tests/test_color_map_selector.py

@@ -0,0 +1,43 @@
+import pyqtgraph as pg
+import pytest
+
+from bec_widgets.widgets.color_map_selector.color_map_selector import ColormapSelector
+
+
+@pytest.fixture
+def color_map_selector(qtbot):
+    widget = ColormapSelector()
+    qtbot.addWidget(widget)
+    qtbot.waitExposed(widget)
+    yield widget
+    widget.close()
+
+
+def test_color_map_selector_init(color_map_selector):
+    assert color_map_selector is not None
+    assert isinstance(color_map_selector, ColormapSelector)
+
+    all_maps = pg.colormap.listMaps()
+    loaded_maps = [
+        color_map_selector.combo.itemText(i) for i in range(color_map_selector.combo.count())
+    ]
+    assert len(loaded_maps) > 0
+    assert all_maps == loaded_maps
+
+
+def test_color_map_selector_add_color_maps(color_map_selector):
+    color_map_selector.add_color_maps(["cividis", "viridis"])
+    assert color_map_selector.combo.count() == 2
+    assert color_map_selector.combo.itemText(0) == "cividis"
+    assert color_map_selector.combo.itemText(1) == "viridis"
+    assert color_map_selector.combo.itemText(2) != "cividis"
+    assert color_map_selector.combo.itemText(2) != "viridis"
+
+
+def test_colormap_add_maps_by_property(color_map_selector):
+    color_map_selector.colormaps = ["cividis", "viridis"]
+    assert color_map_selector.combo.count() == 2
+    assert color_map_selector.combo.itemText(0) == "cividis"
+    assert color_map_selector.combo.itemText(1) == "viridis"
+    assert color_map_selector.combo.itemText(2) != "cividis"
+    assert color_map_selector.combo.itemText(2) != "viridis"

tests/unit_tests/test_device_input_widgets.py

@@ -67,6 +67,7 @@ def test_device_input_combobox_init(device_input_combobox):
         "bpm3a",
         "bpm3i",
         "eiger",
+        "async_device",
         "test",
     ]
 
@@ -154,6 +155,7 @@ def test_device_input_line_edit_init(device_input_line_edit):
         "bpm3a",
         "bpm3i",
         "eiger",
+        "async_device",
         "test",
     ]
 

tests/unit_tests/test_waveform1d.py

@@ -14,36 +14,36 @@ def test_adding_curve_to_waveform(bec_figure):
     w1 = bec_figure.plot()
 
     # adding curve which is in bec - only names
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="bpm4i")
     assert c1.config.label == "bpm4i-bpm4i"
 
     # adding curve which is in bec - names and entry
-    c2 = w1.add_curve_scan(x_name="samx", x_entry="samx", y_name="bpm3a", y_entry="bpm3a")
+    c2 = w1.add_curve_bec(x_name="samx", x_entry="samx", y_name="bpm3a", y_entry="bpm3a")
     assert c2.config.label == "bpm3a-bpm3a"
 
     # adding curve which is not in bec
     with pytest.raises(ValueError) as excinfo:
-        w1.add_curve_scan(x_name="non_existent_device", y_name="non_existent_device")
+        w1.add_curve_bec(x_name="non_existent_device", y_name="non_existent_device")
     assert "Device 'non_existent_device' not found in current BEC session" in str(excinfo.value)
 
     # adding wrong entry for samx
     with pytest.raises(ValueError) as excinfo:
-        w1.add_curve_scan(
+        w1.add_curve_bec(
             x_name="samx", x_entry="non_existent_entry", y_name="bpm3a", y_entry="bpm3a"
         )
     assert "Entry 'non_existent_entry' not found in device 'samx' signals" in str(excinfo.value)
 
     # adding wrong device with validation switched off
-    c3 = w1.add_curve_scan(x_name="samx", y_name="non_existent_device", validate_bec=False)
+    c3 = w1.add_curve_bec(x_name="samx", y_name="non_existent_device", validate_bec=False)
     assert c3.config.label == "non_existent_device-non_existent_device"
 
 
 def test_adding_curve_with_same_id(bec_figure):
     w1 = bec_figure.plot()
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i", gui_id="test_curve")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="bpm4i", gui_id="test_curve")
 
     with pytest.raises(ValueError) as excinfo:
-        w1.add_curve_scan(x_name="samx", y_name="bpm4i", gui_id="test_curve")
+        w1.add_curve_bec(x_name="samx", y_name="bpm4i", gui_id="test_curve")
         assert "Curve with ID 'test_curve' already exists." in str(excinfo.value)
 
 
@@ -134,7 +134,7 @@ def test_create_waveform1D_by_config(bec_figure):
 
 def test_change_gui_id(bec_figure):
     w1 = bec_figure.plot()
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="bpm4i")
     w1.change_gui_id("new_id")
 
     assert w1.config.gui_id == "new_id"
@@ -143,7 +143,7 @@ def test_change_gui_id(bec_figure):
 
 def test_getting_curve(bec_figure):
     w1 = bec_figure.plot()
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i", gui_id="test_curve")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="bpm4i", gui_id="test_curve")
     c1_expected_config = CurveConfig(
         widget_class="BECCurve",
         gui_id="test_curve",
@@ -166,6 +166,8 @@ def test_getting_curve(bec_figure):
     assert w1.curves[0].config == c1_expected_config
     assert w1._curves_data["scan_segment"]["bpm4i-bpm4i"].config == c1_expected_config
     assert w1.get_curve(0).config == c1_expected_config
+    assert w1.get_curve_config("bpm4i-bpm4i", dict_output=True) == c1_expected_config.model_dump()
+    assert w1.get_curve_config("bpm4i-bpm4i", dict_output=False) == c1_expected_config
     assert w1.get_curve("bpm4i-bpm4i").config == c1_expected_config
     assert c1.get_config(False) == c1_expected_config
     assert c1.get_config() == c1_expected_config.model_dump()
@@ -173,7 +175,7 @@ def test_getting_curve(bec_figure):
 
 def test_getting_curve_errors(bec_figure):
     w1 = bec_figure.plot()
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i", gui_id="test_curve")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="bpm4i", gui_id="test_curve")
 
     with pytest.raises(ValueError) as excinfo:
         w1.get_curve("non_existent_curve")
@@ -191,7 +193,7 @@ def test_getting_curve_errors(bec_figure):
 def test_add_curve(bec_figure):
     w1 = bec_figure.plot()
 
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="bpm4i")
 
     assert len(w1.curves) == 1
     assert w1._curves_data["scan_segment"] == {"bpm4i-bpm4i": c1}
@@ -202,7 +204,7 @@ def test_add_curve(bec_figure):
 def test_change_legend_font_size(bec_figure):
     plot = bec_figure.plot()
 
-    w1 = plot.add_curve_scan(x_name="samx", y_name="bpm4i")
+    w1 = plot.add_curve_bec(x_name="samx", y_name="bpm4i")
     my_func = plot.plot_item.legend
     with mock.patch.object(my_func, "setScale") as mock_set_scale:
         plot.set_legend_label_size(18)
@@ -214,8 +216,8 @@ def test_change_legend_font_size(bec_figure):
 def test_remove_curve(bec_figure):
     w1 = bec_figure.plot()
 
-    w1.add_curve_scan(x_name="samx", y_name="bpm4i")
-    w1.add_curve_scan(x_name="samx", y_name="bpm3a")
+    w1.add_curve_bec(x_name="samx", y_name="bpm4i")
+    w1.add_curve_bec(x_name="samx", y_name="bpm3a")
     w1.remove_curve(0)
     w1.remove_curve("bpm3a-bpm3a")
 
@@ -232,7 +234,7 @@ def test_remove_curve(bec_figure):
 def test_change_curve_appearance_methods(bec_figure, qtbot):
     w1 = bec_figure.plot()
 
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="bpm4i")
 
     c1.set_color("#0000ff")
     c1.set_symbol("x")
@@ -261,7 +263,7 @@ def test_change_curve_appearance_methods(bec_figure, qtbot):
 def test_change_curve_appearance_args(bec_figure):
     w1 = bec_figure.plot()
 
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="bpm4i")
 
     c1.set(
         color="#0000ff",
@@ -414,7 +416,7 @@ def test_curve_add_by_config(bec_figure):
 def test_scan_update(bec_figure, qtbot):
     w1 = bec_figure.plot()
 
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="bpm4i")
 
     msg_waveform = {
         "data": {
@@ -448,7 +450,7 @@ def test_scan_update(bec_figure, qtbot):
 def test_scan_history_with_val_access(bec_figure, qtbot):
     w1 = bec_figure.plot()
 
-    c1 = w1.add_curve_scan(x_name="samx", y_name="bpm4i")
+    w1.plot(x_name="samx", y_name="bpm4i")
 
     mock_scan_data = {
         "samx": {"samx": mock.MagicMock(val=np.array([1, 2, 3]))},  # Use mock.MagicMock for .val
@@ -464,7 +466,7 @@ def test_scan_history_with_val_access(bec_figure, qtbot):
 
     qtbot.wait(500)
 
-    x_data, y_data = c1.get_data()
+    x_data, y_data = w1.curves[0].get_data()
 
     assert np.array_equal(x_data, [1, 2, 3])
     assert np.array_equal(y_data, [4, 5, 6])
@@ -473,7 +475,7 @@ def test_scan_history_with_val_access(bec_figure, qtbot):
 def test_scatter_2d_update(bec_figure, qtbot):
     w1 = bec_figure.plot()
 
-    c1 = w1.add_curve_scan(x_name="samx", y_name="samx", z_name="bpm4i")
+    c1 = w1.add_curve_bec(x_name="samx", y_name="samx", z_name="bpm4i")
 
     msg = {
         "data": {
@@ -485,8 +487,8 @@ def test_scatter_2d_update(bec_figure, qtbot):
     }
     msg_metadata = {"scan_name": "line_scan"}
 
-    mock_scan_data = mock.MagicMock()
-    mock_scan_data.data = {
+    mock_scan_item = mock.MagicMock()
+    mock_scan_item.data = {
         device_name: {
             entry: mock.MagicMock(val=msg["data"][device_name][entry]["value"])
             for entry in msg["data"][device_name]
@@ -494,7 +496,7 @@ def test_scatter_2d_update(bec_figure, qtbot):
         for device_name in msg["data"]
     }
 
-    w1.queue.scan_storage.find_scan_by_ID.return_value = mock_scan_data
+    w1.queue.scan_storage.find_scan_by_ID.return_value = mock_scan_item
 
     w1.on_scan_segment(msg, msg_metadata)
     qtbot.wait(500)
@@ -508,3 +510,180 @@ def test_scatter_2d_update(bec_figure, qtbot):
 
     assert np.array_equal(data, expected_x_y_data)
     assert colors == expected_z_colors
+
+
+def test_waveform_single_arg_inputs(bec_figure, qtbot):
+    w1 = bec_figure.plot()
+
+    w1.plot("bpm4i")
+    w1.plot([1, 2, 3], label="just_y")
+    w1.plot([3, 4, 5], [7, 8, 9], label="x_y")
+    w1.plot(x=[1, 2, 3], y=[4, 5, 6], label="x_y_kwargs")
+    data_array_1D = np.random.rand(10)
+    data_array_2D = np.random.rand(10, 2)
+    w1.plot(data_array_1D, label="np_ndarray 1D")
+    w1.plot(data_array_2D, label="np_ndarray 2D")
+
+    qtbot.wait(200)
+
+    assert w1._curves_data["scan_segment"]["bpm4i-bpm4i"].config.label == "bpm4i-bpm4i"
+    assert w1._curves_data["custom"]["just_y"].config.label == "just_y"
+    assert w1._curves_data["custom"]["x_y"].config.label == "x_y"
+    assert w1._curves_data["custom"]["x_y_kwargs"].config.label == "x_y_kwargs"
+
+    assert np.array_equal(w1._curves_data["custom"]["just_y"].get_data(), ([0, 1, 2], [1, 2, 3]))
+    assert np.array_equal(w1._curves_data["custom"]["just_y"].get_data(), ([0, 1, 2], [1, 2, 3]))
+    assert np.array_equal(w1._curves_data["custom"]["x_y"].get_data(), ([3, 4, 5], [7, 8, 9]))
+    assert np.array_equal(
+        w1._curves_data["custom"]["x_y_kwargs"].get_data(), ([1, 2, 3], [4, 5, 6])
+    )
+    assert np.array_equal(
+        w1._curves_data["custom"]["np_ndarray 1D"].get_data(),
+        (np.arange(data_array_1D.size), data_array_1D.T),
+    )
+    assert np.array_equal(w1._curves_data["custom"]["np_ndarray 2D"].get_data(), data_array_2D.T)
+
+
+def test_waveform_set_x_sync(bec_figure, qtbot):
+    w1 = bec_figure.plot()
+    custom_label = "custom_label"
+    w1.plot("bpm4i")
+    w1.set_x_label(custom_label)
+
+    scan_item_mock = mock.MagicMock()
+    mock_data = {
+        "samx": {"samx": mock.MagicMock(val=np.array([1, 2, 3]))},
+        "samy": {"samy": mock.MagicMock(val=np.array([4, 5, 6]))},
+        "bpm4i": {
+            "bpm4i": mock.MagicMock(
+                val=np.array([7, 8, 9]),
+                timestamps=np.array([1720520189.959115, 1720520189.986618, 1720520190.0157812]),
+            )
+        },
+    }
+
+    scan_item_mock.data = mock_data
+    scan_item_mock.status_message.info = {"scan_report_devices": ["samx"]}
+
+    w1.queue.scan_storage.find_scan_by_ID.return_value = scan_item_mock
+
+    w1.on_scan_segment({"scan_id": 1}, {})
+    qtbot.wait(200)
+
+    # Best effort - samx
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.array_equal(x_data, [1, 2, 3])
+    assert np.array_equal(y_data, [7, 8, 9])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [auto: samx-samx]"
+
+    # Change to samy
+    w1.set_x("samy")
+    qtbot.wait(200)
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.array_equal(x_data, [4, 5, 6])
+    assert np.array_equal(y_data, [7, 8, 9])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [samy-samy]"
+
+    # change to index
+    w1.set_x("index")
+    qtbot.wait(200)
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.array_equal(x_data, [0, 1, 2])
+    assert np.array_equal(y_data, [7, 8, 9])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [index]"
+
+    # change to timestamp
+    w1.set_x("timestamp")
+    qtbot.wait(200)
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.allclose(x_data, np.array([1.72052019e09, 1.72052019e09, 1.72052019e09]))
+    assert np.array_equal(y_data, [7, 8, 9])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [timestamp]"
+
+
+def test_waveform_async_data_update(bec_figure, qtbot):
+    w1 = bec_figure.plot("async_device")
+    custom_label = "custom_label"
+    w1.set_x_label(custom_label)
+
+    # scan_item_mock = mock.MagicMock()
+    # mock_data = {
+    #     "async_device": {
+    #         "async_device": mock.MagicMock(
+    #             val=np.array([7, 8, 9]),
+    #             timestamps=np.array([1720520189.959115, 1720520189.986618, 1720520190.0157812]),
+    #         )
+    #     }
+    # }
+    #
+    # scan_item_mock.async_data = mock_data
+    # w1.queue.scan_storage.find_scan_by_ID.return_value = scan_item_mock
+
+    msg_1 = {"signals": {"async_device": {"value": [7, 8, 9]}}}
+    w1.on_async_readback(msg_1, {"async_update": "extend"})
+
+    qtbot.wait(200)
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.array_equal(x_data, [0, 1, 2])
+    assert np.array_equal(y_data, [7, 8, 9])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [best_effort]"
+
+    msg_2 = {"signals": {"async_device": {"value": [10, 11, 12]}}}
+    w1.on_async_readback(msg_2, {"async_update": "extend"})
+
+    qtbot.wait(200)
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.array_equal(x_data, [0, 1, 2, 3, 4, 5])
+    assert np.array_equal(y_data, [7, 8, 9, 10, 11, 12])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [best_effort]"
+
+    msg_3 = {"signals": {"async_device": {"value": [20, 21, 22]}}}
+    w1.on_async_readback(msg_3, {"async_update": "replace"})
+
+    qtbot.wait(200)
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.array_equal(x_data, [0, 1, 2])
+    assert np.array_equal(y_data, [20, 21, 22])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [best_effort]"
+
+
+def test_waveform_set_x_async(bec_figure, qtbot):
+    w1 = bec_figure.plot("async_device")
+    custom_label = "custom_label"
+    w1.set_x_label(custom_label)
+
+    scan_item_mock = mock.MagicMock()
+    mock_data = {
+        "async_device": {
+            "async_device": {
+                "value": np.array([7, 8, 9]),
+                "timestamp": np.array([1720520189.959115, 1720520189.986618, 1720520190.0157812]),
+            }
+        }
+    }
+
+    scan_item_mock.async_data = mock_data
+    w1.queue.scan_storage.find_scan_by_ID.return_value = scan_item_mock
+
+    w1.on_scan_status({"scan_id": 1})
+    w1.replot_async_curve()
+
+    qtbot.wait(200)
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.array_equal(x_data, [0, 1, 2])
+    assert np.array_equal(y_data, [7, 8, 9])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [best_effort]"
+
+    w1.set_x("timestamp")
+    qtbot.wait(200)
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.allclose(x_data, np.array([1.72052019e09, 1.72052019e09, 1.72052019e09]))
+    assert np.array_equal(y_data, [7, 8, 9])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [timestamp]"
+
+    w1.set_x("index")
+    qtbot.wait(200)
+    x_data, y_data = w1.curves[0].get_data()
+    assert np.array_equal(x_data, [0, 1, 2])
+    assert np.array_equal(y_data, [7, 8, 9])
+    assert w1.plot_item.getAxis("bottom").labelText == custom_label + " [index]"