add checks for order_of_images

Author: bendichter

nwbinspector/checks/images.py

@@ -0,0 +1,26 @@
+from hdmf.utils import get_data_shape
+
+from pynwb.base import Images
+
+from ..register_checks import register_check, Importance, InspectorMessage
+
+
+@register_check(importance=Importance.BEST_PRACTICE_VIOLATION, neurodata_type=Images)
+def check_order_of_images_unique(images: Images):
+    if images.order_of_images is None:
+        return
+    if not len(set(images.order_of_images)) == len(images.order_of_images):
+        return InspectorMessage(
+            message="order_of_images should have unique values."
+        )
+
+
+@register_check(importance=Importance.BEST_PRACTICE_VIOLATION, neurodata_type=Images)
+def check_order_of_images_len(images: Images):
+    if images.order_of_images is None:
+        return
+    if not len(images.order_of_images) == len(images.images):
+        return InspectorMessage(
+            message=f"Length of order_of_images ({len(images.order_of_images)}) does not match the number of images ("
+                    f"{len(images.images)})."
+        )

tests/unit_tests/test_images.py

@@ -0,0 +1,48 @@
+import numpy as np
+
+from pynwb.base import Images, ImageReferences
+from pynwb.image import GrayscaleImage
+
+from nwbinspector.checks.images import check_order_of_images_unique, check_order_of_images_len
+
+
+def test_check_order_of_images_unique():
+
+    imgs = [GrayscaleImage(name=f"image{i}", data=np.random.randn(10, 10)) for i in range(5)]
+    img_refs = ImageReferences(name="order_of_images", data=imgs + [imgs[0]])
+    images = Images(name="my_images", images=imgs, order_of_images=img_refs)
+
+    assert check_order_of_images_unique(images).message == "order_of_images should have unique values."
+
+
+def test_pass_check_order_of_images_unique():
+
+    imgs = [GrayscaleImage(name=f"image{i}", data=np.random.randn(10, 10)) for i in range(5)]
+    img_refs = ImageReferences(name="order_of_images", data=imgs)
+    images = Images(name="my_images", images=imgs, order_of_images=img_refs)
+
+    assert check_order_of_images_unique(images) is None
+
+
+def test_check_order_of_images_len():
+
+    imgs = [GrayscaleImage(name=f"image{i}", data=np.random.randn(10, 10)) for i in range(5)]
+    img_refs = ImageReferences(name="order_of_images", data=imgs + [imgs[0]])
+    images = Images(name="my_images", images=imgs, order_of_images=img_refs)
+
+    assert (
+        check_order_of_images_len(images).message
+        == f"Length of order_of_images (6) does not match the number of images (5)."
+    )
+
+
+def test_pass_check_order_of_images_len():
+
+    imgs = [GrayscaleImage(name=f"image{i}", data=np.random.randn(10, 10)) for i in range(5)]
+    img_refs = ImageReferences(name="order_of_images", data=imgs)
+    images = Images(name="my_images", images=imgs, order_of_images=img_refs)
+
+    assert check_order_of_images_len(images) is None
+
+
+