monai-weekly 1.5.dev2506__py3-none-any.whl → 1.5.dev2508__py3-none-any.whl

tests/apps/pathology/transforms/post/test_hovernet_instance_map_post_processing.py

@@ -0,0 +1,61 @@
+# Copyright (c) MONAI Consortium
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#     http://www.apache.org/licenses/LICENSE-2.0
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+import unittest
+
+import numpy as np
+from parameterized import parameterized
+
+from monai.apps.pathology.transforms.post.array import HoVerNetInstanceMapPostProcessing
+from monai.transforms import ComputeHoVerMaps, FillHoles, GaussianSmooth
+from monai.utils import min_version, optional_import
+from tests.test_utils import TEST_NDARRAYS, assert_allclose
+
+_, has_scipy = optional_import("scipy", "1.8.1", min_version)
+_, has_skimage = optional_import("skimage", "0.19.3", min_version)
+
+y, x = np.ogrid[0:30, 0:30]
+image = (x - 10) ** 2 + (y - 10) ** 2 <= 5**2
+image = image[None, ...].astype("uint8")
+
+TEST_CASE_1 = [{}, {"1": {"centroid": 1, "bbox": 1.0}}, np.zeros_like(image)]
+TEST_CASE_2 = [{"distance_smooth_fn": GaussianSmooth()}, {"1": {"type": 1, "type_prob": 1.0}}, np.zeros_like(image)]
+TEST_CASE_3 = [{"marker_postprocess_fn": FillHoles()}, {"1": {"type": 1, "type_prob": 1.0}}, np.zeros_like(image)]
+
+TEST_CASE = []
+for p in TEST_NDARRAYS:
+    TEST_CASE.append([p, image] + TEST_CASE_1)
+    TEST_CASE.append([p, image] + TEST_CASE_2)
+    TEST_CASE.append([p, image] + TEST_CASE_3)
+
+
+@unittest.skipUnless(has_scipy, "Requires scipy library.")
+@unittest.skipUnless(has_skimage, "Requires scikit-image library.")
+class TestHoVerNetInstanceMapPostProcessing(unittest.TestCase):
+    @parameterized.expand(TEST_CASE)
+    def test_value(self, in_type, test_data, kwargs, expected_info, expected_map):
+        nuclear_prediction = in_type(test_data.astype(float))
+        hover_map = in_type(ComputeHoVerMaps()(test_data.astype(int)))
+
+        inst_info, inst_map = HoVerNetInstanceMapPostProcessing(**kwargs)(nuclear_prediction, hover_map)
+
+        # instance info
+        for key in inst_info:
+            assert_allclose(inst_info[key]["centroid"], expected_info[key]["centroid"], type_test=False)
+
+        # instance map
+        assert_allclose(inst_map, expected_map, type_test=False)
+
+
+if __name__ == "__main__":
+    unittest.main()

tests/apps/pathology/transforms/post/test_hovernet_instance_map_post_processingd.py

@@ -0,0 +1,66 @@
+# Copyright (c) MONAI Consortium
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#     http://www.apache.org/licenses/LICENSE-2.0
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+import unittest
+
+import numpy as np
+from parameterized import parameterized
+
+from monai.apps.pathology.transforms.post.dictionary import HoVerNetInstanceMapPostProcessingd
+from monai.transforms import ComputeHoVerMaps, FillHoles, GaussianSmooth
+from monai.utils import min_version, optional_import
+from monai.utils.enums import HoVerNetBranch
+from tests.test_utils import TEST_NDARRAYS, assert_allclose
+
+_, has_scipy = optional_import("scipy", "1.8.1", min_version)
+_, has_skimage = optional_import("skimage", "0.19.3", min_version)
+
+y, x = np.ogrid[0:30, 0:30]
+image = (x - 10) ** 2 + (y - 10) ** 2 <= 5**2
+image = image[None, ...].astype("uint8")
+
+TEST_CASE_1 = [{}, {"1": {"centroid": 1, "bbox": 1.0}}, np.zeros_like(image)]
+TEST_CASE_2 = [{"distance_smooth_fn": GaussianSmooth()}, {"1": {"type": 1, "type_prob": 1.0}}, np.zeros_like(image)]
+TEST_CASE_3 = [{"marker_postprocess_fn": FillHoles()}, {"1": {"type": 1, "type_prob": 1.0}}, np.zeros_like(image)]
+
+TEST_CASE = []
+for p in TEST_NDARRAYS:
+    TEST_CASE.append([p, image] + TEST_CASE_1)
+    TEST_CASE.append([p, image] + TEST_CASE_2)
+    TEST_CASE.append([p, image] + TEST_CASE_3)
+
+
+@unittest.skipUnless(has_scipy, "Requires scipy library.")
+@unittest.skipUnless(has_skimage, "Requires scikit-image library.")
+class TestHoVerNetInstanceMapPostProcessingd(unittest.TestCase):
+    @parameterized.expand(TEST_CASE)
+    def test_value(self, in_type, test_data, kwargs, expected_info, expected_map):
+        input = {
+            HoVerNetBranch.NP.value: in_type(test_data.astype(float)),
+            HoVerNetBranch.HV.value: in_type(ComputeHoVerMaps()(test_data.astype(int))),
+        }
+
+        outputs = HoVerNetInstanceMapPostProcessingd(**kwargs)(input)
+        inst_info_key = kwargs.get("instance_info_key", "instance_info")
+        inst_map_key = kwargs.get("instance_map_key", "instance_map")
+
+        # instance info
+        for key in outputs[inst_info_key]:
+            assert_allclose(outputs[inst_info_key]["centroid"], expected_info[key]["centroid"], type_test=False)
+
+        # instance map
+        assert_allclose(outputs[inst_map_key], expected_map, type_test=False)
+
+
+if __name__ == "__main__":
+    unittest.main()

tests/apps/pathology/transforms/post/test_hovernet_nuclear_type_post_processing.py

@@ -0,0 +1,66 @@
+# Copyright (c) MONAI Consortium
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#     http://www.apache.org/licenses/LICENSE-2.0
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+import unittest
+
+import numpy as np
+from parameterized import parameterized
+
+from monai.apps.pathology.transforms.post.array import (
+    HoVerNetInstanceMapPostProcessing,
+    HoVerNetNuclearTypePostProcessing,
+)
+from monai.transforms import ComputeHoVerMaps
+from monai.utils import min_version, optional_import
+from tests.test_utils import TEST_NDARRAYS, assert_allclose
+
+_, has_scipy = optional_import("scipy", "1.8.1", min_version)
+_, has_skimage = optional_import("skimage", "0.19.3", min_version)
+
+y, x = np.ogrid[0:30, 0:30]
+image = (x - 10) ** 2 + (y - 10) ** 2 <= 5**2
+image = image[None, ...].astype("uint8")
+
+TEST_CASE_1 = [{}, {"1": {"type": 1, "type_prob": 1.0}}, np.zeros_like(image)]
+
+TEST_CASE = []
+for p in TEST_NDARRAYS:
+    TEST_CASE.append([p, image] + TEST_CASE_1)
+
+
+@unittest.skipUnless(has_scipy, "Requires scipy library.")
+@unittest.skipUnless(has_skimage, "Requires scikit-image library.")
+class TestHoVerNetNuclearTypePostProcessing(unittest.TestCase):
+    @parameterized.expand(TEST_CASE)
+    def test_value(self, in_type, test_data, kwargs, expected_info, expected_map):
+        nuclear_prediction = in_type(test_data.astype(float))
+        hover_map = in_type(ComputeHoVerMaps()(test_data.astype(int)))
+        nuclear_type = in_type(test_data)
+
+        inst_info, inst_map = HoVerNetInstanceMapPostProcessing()(nuclear_prediction, hover_map)
+        inst_info, type_map = HoVerNetNuclearTypePostProcessing(**kwargs)(nuclear_type, inst_info, inst_map)
+
+        # instance prediction info
+        for key in inst_info:
+            self.assertEqual(inst_info[key]["type"], expected_info[key]["type"])
+            self.assertEqual(inst_info[key]["type_prob"], expected_info[key]["type_prob"])
+
+        # type map
+        if expected_map is None:
+            self.assertIsNone(type_map)
+        else:
+            assert_allclose(type_map, expected_map, type_test=False)
+
+
+if __name__ == "__main__":
+    unittest.main()

tests/apps/pathology/transforms/post/test_watershed.py

@@ -0,0 +1,60 @@
+# Copyright (c) MONAI Consortium
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#     http://www.apache.org/licenses/LICENSE-2.0
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+import unittest
+
+import numpy as np
+from parameterized import parameterized
+
+from monai.apps.pathology.transforms.post.array import (
+    GenerateDistanceMap,
+    GenerateInstanceBorder,
+    GenerateWatershedMarkers,
+    GenerateWatershedMask,
+    Watershed,
+)
+from monai.utils import min_version, optional_import
+from tests.test_utils import TEST_NDARRAYS
+
+_, has_skimage = optional_import("skimage", "0.19.3", min_version)
+_, has_scipy = optional_import("scipy", "1.8.1", min_version)
+
+np.random.RandomState(123)
+
+TESTS = []
+params = {"connectivity": 1}
+for p in TEST_NDARRAYS:
+    image = p(np.random.rand(1, 10, 10))
+    hover_map = p(np.random.rand(2, 10, 10))
+
+    TESTS.append([params, image, hover_map, (1, 10, 10)])
+
+
+@unittest.skipUnless(has_skimage, "Requires scikit-image library.")
+@unittest.skipUnless(has_scipy, "Requires scipy library.")
+class TestWatershed(unittest.TestCase):
+    @parameterized.expand(TESTS)
+    def test_output(self, args, image, hover_map, expected_shape):
+        mask = GenerateWatershedMask()(image)
+        border_map = GenerateInstanceBorder(kernel_size=3)(mask, hover_map)
+        distance_map = GenerateDistanceMap()(mask, border_map)
+        markers = GenerateWatershedMarkers()(mask, border_map)
+
+        calculate_instance_seg = Watershed(**args)
+        output = calculate_instance_seg(distance_map, mask, markers)
+
+        self.assertTupleEqual(output.shape, expected_shape)
+
+
+if __name__ == "__main__":
+    unittest.main()

tests/apps/pathology/transforms/post/test_watershedd.py

@@ -0,0 +1,70 @@
+# Copyright (c) MONAI Consortium
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#     http://www.apache.org/licenses/LICENSE-2.0
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+import unittest
+
+import numpy as np
+from parameterized import parameterized
+
+from monai.apps.pathology.transforms.post.dictionary import (
+    GenerateDistanceMapd,
+    GenerateInstanceBorderd,
+    GenerateWatershedMarkersd,
+    GenerateWatershedMaskd,
+    Watershedd,
+)
+from monai.transforms import Compose
+from monai.utils import min_version, optional_import
+from tests.test_utils import TEST_NDARRAYS
+
+_, has_skimage = optional_import("skimage", "0.19.3", min_version)
+_, has_scipy = optional_import("scipy", "1.8.1", min_version)
+
+TESTS = []
+params = {"keys": "dist_map", "mask_key": "mask", "markers_key": "markers", "connectivity": 1}
+for p in TEST_NDARRAYS:
+    image = p(np.random.rand(1, 10, 10))
+    hover_map = p(np.random.rand(2, 10, 10))
+
+    TESTS.append([params, image, hover_map, (1, 10, 10)])
+
+    params.update({"markers_key": None})
+    TESTS.append([params, image, hover_map, (1, 10, 10)])
+
+    params.update({"mask_key": None, "markers_key": None})
+    TESTS.append([params, image, hover_map, (1, 10, 10)])
+
+
+@unittest.skipUnless(has_skimage, "Requires scikit-image library.")
+@unittest.skipUnless(has_scipy, "Requires scipy library.")
+class TestWatershedd(unittest.TestCase):
+    @parameterized.expand(TESTS)
+    def test_output(self, args, image, hover_map, expected_shape):
+        data = {"output": image, "hover_map": hover_map}
+
+        trans = Compose(
+            [
+                GenerateWatershedMaskd(keys="output"),
+                GenerateInstanceBorderd(mask_key="mask", hover_map_key="hover_map", kernel_size=3),
+                GenerateDistanceMapd(mask_key="mask", border_key="border"),
+                GenerateWatershedMarkersd(mask_key="mask", border_key="border"),
+                Watershedd(**args),
+            ]
+        )
+
+        output = trans(data)
+        self.assertTupleEqual(output["dist_map"].shape, expected_shape)
+
+
+if __name__ == "__main__":
+    unittest.main()

tests/apps/pathology/transforms/test_pathology_he_stain.py

@@ -0,0 +1,230 @@
+# Copyright (c) MONAI Consortium
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#     http://www.apache.org/licenses/LICENSE-2.0
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+import unittest
+
+import numpy as np
+from parameterized import parameterized
+
+from monai.apps.pathology.transforms import ExtractHEStains, NormalizeHEStains
+
+# None inputs
+EXTRACT_STAINS_TEST_CASE_0 = (None,)
+EXTRACT_STAINS_TEST_CASE_00 = (None, None)
+NORMALIZE_STAINS_TEST_CASE_0 = (None,)
+NORMALIZE_STAINS_TEST_CASE_00: tuple = ({}, None, None)
+
+# input pixels with negative values
+NEGATIVE_VALUE_TEST_CASE = [np.full((3, 2, 3), -1)]
+
+# input pixels with greater than 255 values
+INVALID_VALUE_TEST_CASE = [np.full((3, 2, 3), 256)]
+
+# input pixels all transparent and below the beta absorbance threshold
+EXTRACT_STAINS_TEST_CASE_1 = [np.full((3, 2, 3), 240)]
+
+# input pixels uniformly filled, but above beta absorbance threshold
+EXTRACT_STAINS_TEST_CASE_2 = [np.full((3, 2, 3), 100)]
+
+# input pixels uniformly filled (different value), but above beta absorbance threshold
+EXTRACT_STAINS_TEST_CASE_3 = [np.full((3, 2, 3), 150)]
+
+# input pixels uniformly filled with zeros, leading to two identical stains extracted
+EXTRACT_STAINS_TEST_CASE_4 = [
+    np.zeros((3, 2, 3)),
+    np.array([[0.0, 0.0], [0.70710678, 0.70710678], [0.70710678, 0.70710678]]),
+]
+
+# input pixels not uniformly filled, leading to two different stains extracted
+EXTRACT_STAINS_TEST_CASE_5 = [
+    np.array([[[100, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0]]]),
+    np.array([[0.70710677, 0.18696113], [0.0, 0.0], [0.70710677, 0.98236734]]),
+]
+
+# input pixels all transparent and below the beta absorbance threshold
+NORMALIZE_STAINS_TEST_CASE_1 = [np.full((3, 2, 3), 240)]
+
+# input pixels uniformly filled with zeros, and target stain matrix provided
+NORMALIZE_STAINS_TEST_CASE_2 = [{"target_he": np.full((3, 2), 1)}, np.zeros((3, 2, 3)), np.full((3, 2, 3), 11)]
+
+# input pixels uniformly filled with zeros, and target stain matrix not provided
+NORMALIZE_STAINS_TEST_CASE_3 = [
+    {},
+    np.zeros((3, 2, 3)),
+    np.array([[[63, 25, 60], [63, 25, 60]], [[63, 25, 60], [63, 25, 60]], [[63, 25, 60], [63, 25, 60]]]),
+]
+
+# input pixels not uniformly filled
+NORMALIZE_STAINS_TEST_CASE_4 = [
+    {"target_he": np.full((3, 2), 1)},
+    np.array([[[100, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0]]]),
+    np.array([[[87, 87, 87], [33, 33, 33]], [[33, 33, 33], [33, 33, 33]], [[33, 33, 33], [33, 33, 33]]]),
+]
+
+
+class TestExtractHEStains(unittest.TestCase):
+
+    @parameterized.expand(
+        [NEGATIVE_VALUE_TEST_CASE, INVALID_VALUE_TEST_CASE, EXTRACT_STAINS_TEST_CASE_0, EXTRACT_STAINS_TEST_CASE_1]
+    )
+    def test_transparent_image(self, image):
+        """
+        Test HE stain extraction on an image that comprises
+        only transparent pixels - pixels with absorbance below the
+        beta absorbance threshold. A ValueError should be raised,
+        since once the transparent pixels are removed, there are no
+        remaining pixels to compute eigenvectors.
+        """
+        if image is None:
+            with self.assertRaises(TypeError):
+                ExtractHEStains()(image)
+        else:
+            with self.assertRaises(ValueError):
+                ExtractHEStains()(image)
+
+    @parameterized.expand([EXTRACT_STAINS_TEST_CASE_0, EXTRACT_STAINS_TEST_CASE_2, EXTRACT_STAINS_TEST_CASE_3])
+    def test_identical_result_vectors(self, image):
+        """
+        Test HE stain extraction on input images that are
+        uniformly filled with pixels that have absorbance above the
+        beta absorbance threshold. Since input image is uniformly filled,
+        the two extracted stains should have the same RGB values. So,
+        we assert that the first column is equal to the second column
+        of the returned stain matrix.
+        """
+        if image is None:
+            with self.assertRaises(TypeError):
+                ExtractHEStains()(image)
+        else:
+            result = ExtractHEStains()(image)
+            np.testing.assert_array_equal(result[:, 0], result[:, 1])
+
+    @parameterized.expand([EXTRACT_STAINS_TEST_CASE_00, EXTRACT_STAINS_TEST_CASE_4, EXTRACT_STAINS_TEST_CASE_5])
+    def test_result_value(self, image, expected_data):
+        """
+        Test that an input image returns an expected stain matrix.
+
+        For test case 4:
+        - a uniformly filled input image should result in
+          eigenvectors [[1,0,0],[0,1,0],[0,0,1]]
+        - phi should be an array containing only values of
+          arctan(1) since the ratio between the eigenvectors
+          corresponding to the two largest eigenvalues is 1
+        - maximum phi and minimum phi should thus be arctan(1)
+        - thus, maximum vector and minimum vector should be
+          [[0],[0.70710677],[0.70710677]]
+        - the resulting extracted stain should be
+          [[0,0],[0.70710678,0.70710678],[0.70710678,0.70710678]]
+
+        For test case 5:
+        - the non-uniformly filled input image should result in
+          eigenvectors [[0,0,1],[1,0,0],[0,1,0]]
+        - maximum phi and minimum phi should thus be 0.785 and
+          0.188 respectively
+        - thus, maximum vector and minimum vector should be
+          [[0.18696113],[0],[0.98236734]] and
+          [[0.70710677],[0],[0.70710677]] respectively
+        - the resulting extracted stain should be
+          [[0.70710677,0.18696113],[0,0],[0.70710677,0.98236734]]
+        """
+        if image is None:
+            with self.assertRaises(TypeError):
+                ExtractHEStains()(image)
+        else:
+            result = ExtractHEStains()(image)
+            np.testing.assert_allclose(result, expected_data)
+
+
+class TestNormalizeHEStains(unittest.TestCase):
+
+    @parameterized.expand(
+        [NEGATIVE_VALUE_TEST_CASE, INVALID_VALUE_TEST_CASE, NORMALIZE_STAINS_TEST_CASE_0, NORMALIZE_STAINS_TEST_CASE_1]
+    )
+    def test_transparent_image(self, image):
+        """
+        Test HE stain normalization on an image that comprises
+        only transparent pixels - pixels with absorbance below the
+        beta absorbance threshold. A ValueError should be raised,
+        since once the transparent pixels are removed, there are no
+        remaining pixels to compute eigenvectors.
+        """
+        if image is None:
+            with self.assertRaises(TypeError):
+                NormalizeHEStains()(image)
+        else:
+            with self.assertRaises(ValueError):
+                NormalizeHEStains()(image)
+
+    @parameterized.expand(
+        [
+            NORMALIZE_STAINS_TEST_CASE_00,
+            NORMALIZE_STAINS_TEST_CASE_2,
+            NORMALIZE_STAINS_TEST_CASE_3,
+            NORMALIZE_STAINS_TEST_CASE_4,
+        ]
+    )
+    def test_result_value(self, arguments, image, expected_data):
+        """
+        Test that an input image returns an expected normalized image.
+
+        For test case 2:
+        - This case tests calling the stain normalizer, after the
+          _deconvolution_extract_conc function. This is because the normalized
+          concentration returned for each pixel is the same as the reference
+          maximum stain concentrations in the case that the image is uniformly
+          filled, as in this test case. This is because the maximum concentration
+          for each stain is the same as each pixel's concentration.
+        - Thus, the normalized concentration matrix should be a (2, 6) matrix
+          with the first row having all values of 1.9705, second row all 1.0308.
+        - Taking the matrix product of the target stain matrix and the concentration
+          matrix, then using the inverse Beer-Lambert transform to obtain the RGB
+          image from the absorbance image, and finally converting to uint8,
+          we get that the stain normalized image should be a matrix of
+          dims (3, 2, 3), with all values 11.
+
+        For test case 3:
+        - This case also tests calling the stain normalizer, after the
+          _deconvolution_extract_conc function returns the image concentration
+          matrix.
+        - As in test case 2, the normalized concentration matrix should be a (2, 6) matrix
+          with the first row having all values of 1.9705, second row all 1.0308.
+        - Taking the matrix product of the target default stain matrix and the concentration
+          matrix, then using the inverse Beer-Lambert transform to obtain the RGB
+          image from the absorbance image, and finally converting to uint8,
+          we get that the stain normalized image should be [[[63, 25, 60], [63, 25, 60]],
+          [[63, 25, 60], [63, 25, 60]], [[63, 25, 60], [63, 25, 60]]]
+
+        For test case 4:
+        - For this non-uniformly filled image, the stain extracted should be
+          [[0.70710677,0.18696113],[0,0],[0.70710677,0.98236734]], as validated for the
+          ExtractHEStains class. Solving the linear least squares problem (since
+          absorbance matrix = stain matrix * concentration matrix), we obtain the concentration
+          matrix that should be [[-0.3101, 7.7508, 7.7508, 7.7508, 7.7508, 7.7508],
+          [5.8022, 0, 0, 0, 0, 0]]
+        - Normalizing the concentration matrix, taking the matrix product of the
+          target stain matrix and the concentration matrix, using the inverse
+          Beer-Lambert transform to obtain the RGB image from the absorbance
+          image, and finally converting to uint8, we get that the stain normalized
+          image should be [[[87, 87, 87], [33, 33, 33]], [[33, 33, 33], [33, 33, 33]],
+          [[33, 33, 33], [33, 33, 33]]]
+        """
+        if image is None:
+            with self.assertRaises(TypeError):
+                NormalizeHEStains()(image)
+        else:
+            result = NormalizeHEStains(**arguments)(image)
+            np.testing.assert_allclose(result, expected_data)
+
+
+if __name__ == "__main__":
+    unittest.main()