mirror of
https://github.com/bec-project/bec_widgets.git
synced 2025-12-31 19:11:18 +01:00
85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
from unittest import mock
|
|
|
|
import pyqtgraph as pg
|
|
import pytest
|
|
|
|
from bec_widgets.widgets.plots.image.image_base import ImageLayerManager
|
|
from bec_widgets.widgets.plots.image.image_item import ImageItem
|
|
|
|
|
|
@pytest.fixture()
|
|
def image_layer_manager():
|
|
"""Fixture to create an instance of ImageLayer."""
|
|
layer = ImageLayerManager(plot_item=mock.MagicMock(spec=pg.PlotItem))
|
|
yield layer
|
|
layer.clear()
|
|
|
|
|
|
def test_image_layer_manager_initialization(image_layer_manager):
|
|
"""Test the initialization of the ImageLayer."""
|
|
assert isinstance(image_layer_manager, ImageLayerManager)
|
|
assert image_layer_manager.plot_item is not None
|
|
|
|
|
|
def test_add_image_layer(image_layer_manager):
|
|
"""Test adding an image layer to the ImageLayerManager."""
|
|
image = ImageItem()
|
|
layer = image_layer_manager.add(name="Test Layer", image=image)
|
|
assert layer.image.zValue() == 0
|
|
|
|
image2 = ImageItem()
|
|
layer2 = image_layer_manager.add(name="Test Layer 2", image=image2)
|
|
assert layer2.image.zValue() == 1
|
|
|
|
image3 = ImageItem()
|
|
layer3 = image_layer_manager.add(name="Test Layer 3", image=image3, z_position="top")
|
|
assert layer3.image.zValue() == 2
|
|
|
|
image4 = ImageItem()
|
|
layer4 = image_layer_manager.add(name="Test Layer 4", image=image4, z_position="bottom")
|
|
assert layer4.image.zValue() == -1
|
|
|
|
|
|
def test_remove_image_layer(image_layer_manager):
|
|
"""Test removing an image layer from the ImageLayerManager."""
|
|
image = ImageItem()
|
|
layer = image_layer_manager.add(name="Test Layer", image=image)
|
|
assert len(image_layer_manager) == 1
|
|
|
|
image_layer_manager.remove(layer)
|
|
assert len(image_layer_manager) == 0
|
|
|
|
|
|
def test_clear_image_layers(image_layer_manager):
|
|
"""Test clearing all image layers from the ImageLayerManager."""
|
|
image = ImageItem()
|
|
layer = image_layer_manager.add(name="Test Layer", image=image)
|
|
assert len(image_layer_manager) == 1
|
|
|
|
image_layer_manager.clear()
|
|
assert len(image_layer_manager) == 0
|
|
|
|
|
|
def test_image_layer_manager_getitem(image_layer_manager):
|
|
"""Test getting an image layer by index."""
|
|
image = ImageItem()
|
|
layer = image_layer_manager.add(name="Test Layer", image=image)
|
|
assert image_layer_manager["Test Layer"] == layer
|
|
|
|
with pytest.raises(TypeError):
|
|
_ = image_layer_manager[1]
|
|
|
|
image_layer_manager.remove("Test Layer")
|
|
assert len(image_layer_manager) == 0
|
|
|
|
|
|
def test_image_layer_iteration(image_layer_manager):
|
|
"""Test iterating over image layers."""
|
|
image = ImageItem()
|
|
layer = image_layer_manager.add(name="Test Layer", image=image)
|
|
assert list(image_layer_manager) == [layer]
|
|
|
|
image2 = ImageItem()
|
|
layer2 = image_layer_manager.add(name="Test Layer 2", image=image2)
|
|
assert list(image_layer_manager) == [layer, layer2]
|