mirror of
https://github.com/bec-project/bec_widgets.git
synced 2025-07-14 03:31:50 +02:00
feat(color_map_selector): added colormap selector with plugin
This commit is contained in:
43
tests/unit_tests/test_color_map_selector.py
Normal file
43
tests/unit_tests/test_color_map_selector.py
Normal file
@ -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"
|
Reference in New Issue
Block a user