Add tests

Author: Czaki

package/PartSegImage/image.py

@@ -47,9 +47,13 @@ class ChannelInfoFull:
     def __post_init__(self):
         if not isinstance(self.color_map, (str, np.ndarray)):
             self.color_map = np.array(self.color_map, dtype=np.uint8)
+        if isinstance(self.color_map, np.ndarray):
+            if self.color_map.dtype != np.uint8:
+                message = f"Colormap as array need to be uint8, not {self.color_map.dtype}"
+                raise ValueError(message)
             if self.color_map.ndim in {1, 2}:
                 if self.color_map.shape[0] not in {3, 4}:
-                    message = f"1d color map need to have 3 or 4 elements (RGB or RGBA), not {self.color_map.shape}"
+                    message = f"Color map need to have 3 or 4 elements (RGB or RGBA), not {self.color_map.shape}"
                     raise ValueError(message)
             else:
                 message = f"Colormap as sequence need to be 1d or 2d array, not {self.color_map.shape}"

package/tests/test_PartSegImage/test_image.py

@@ -2,10 +2,11 @@
 import os
 
 import numpy as np
+import numpy.testing as npt
 import pytest
 from skimage.morphology import diamond
 
-from PartSegImage import Channel, ChannelInfo, Image, ImageWriter, TiffImageReader
+from PartSegImage import Channel, ChannelInfo, ChannelInfoFull, Image, ImageWriter, TiffImageReader
 from PartSegImage.image import FRAME_THICKNESS, _hex_to_rgb, _name_to_rgb
 
 
@@ -728,3 +729,44 @@ def test_name_to_rgb_vispy():
     # This test check mapping not defined in fallback dictionary
     pytest.importorskip("vispy", reason="vispy not installed")
     assert _name_to_rgb("lime") == (0, 255, 0)
+
+
+class TestChannelInfoFull:
+    RED_DEF = (255, 0, 0)
+    RED_2D_REF = [[0, 255], [0, 0], [0, 0]]
+
+    def test_str(self):
+        obj = ChannelInfoFull(name="test", color_map="gray", contrast_limits=(0, 20))
+        assert obj.name == "test"
+        assert obj.color_map == "gray"
+        assert obj.contrast_limits == (0, 20)
+
+    @pytest.mark.parametrize("color_map", [tuple(RED_DEF), np.array(RED_DEF, dtype=np.uint8), list(RED_DEF)])
+    def test_1d_colormap(self, color_map):
+        obj = ChannelInfoFull(name="test", color_map=color_map, contrast_limits=(0, 20))
+        assert obj.name == "test"
+        assert np.all(obj.contrast_limits == (0, 20))
+        assert isinstance(obj.color_map, np.ndarray)
+        assert obj.color_map.shape == (3,)
+        npt.assert_array_equal(obj.color_map, self.RED_DEF)
+
+    @pytest.mark.parametrize("color_map", [tuple(RED_2D_REF), np.array(RED_2D_REF, dtype=np.uint8), list(RED_2D_REF)])
+    def test_2d_colormap(self, color_map):
+        obj = ChannelInfoFull(name="test", color_map=color_map, contrast_limits=(0, 20))
+        assert obj.name == "test"
+        assert np.all(obj.contrast_limits == (0, 20))
+        assert isinstance(obj.color_map, np.ndarray)
+        assert obj.color_map.shape == (3, 2)
+        npt.assert_array_equal(obj.color_map, self.RED_2D_REF)
+
+    def test_wrong_dtype_colormap(self):
+        with pytest.raises(ValueError, match="Colormap as array need to be uint8"):
+            ChannelInfoFull(name="test", color_map=np.array(self.RED_DEF, dtype=np.float32), contrast_limits=(0, 20))
+
+    def test_wrong_colors_colormap(self):
+        with pytest.raises(ValueError, match="Color map need to have 3 or 4 elements"):
+            ChannelInfoFull(name="test", color_map=np.array([[0, 0], [0, 0]], dtype=np.uint8), contrast_limits=(0, 20))
+
+    def test_wrong_dimesnsion_colormap(self):
+        with pytest.raises(ValueError, match="Colormap as sequence need to"):
+            ChannelInfoFull(name="test", color_map=np.array([[[0]], [[0]]], dtype=np.uint8), contrast_limits=(0, 20))