From 066edd2295d251c4f3559e7dbf17c708106bf0ab Mon Sep 17 00:00:00 2001
From: Wouter-Michiel Vierdag <michiel.vierdag@embl.de>
Date: Fri, 8 Nov 2024 23:21:37 +0100
Subject: [PATCH] add test for decorator

---
 test/test_utils.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 test/test_utils.py

diff --git a/test/test_utils.py b/test/test_utils.py
new file mode 100644
index 0000000..57828c5
--- /dev/null
+++ b/test/test_utils.py
@@ -0,0 +1,24 @@
+from multiscale_spatial_image import skip_non_dimension_nodes
+import numpy as np
+from spatial_image import to_spatial_image
+from multiscale_spatial_image import to_multiscale
+
+
+def test_skip_nodes():
+    data = np.zeros((2, 200, 200))
+    dims = ("c", "y", "x")
+    scale_factors = [2, 2]
+    image = to_spatial_image(array_like=data, dims=dims)
+    multiscale_img = to_multiscale(image, scale_factors=scale_factors)
+
+    @skip_non_dimension_nodes
+    def transpose(ds, *args, **kwargs):
+        return ds.transpose(*args, **kwargs)
+
+    for scale in list(multiscale_img.keys()):
+        assert multiscale_img[scale]["image"].dims == ("c", "y", "x")
+
+    # applying this function without skipping the root node would fail as the root node does not have dimensions.
+    result = multiscale_img.map_over_datasets(transpose, "y", "x", "c")
+    for scale in list(result.keys()):
+        assert result[scale]["image"].dims == ("y", "x", "c")