suspect that eagle() is somehow getting loaded as RGB instead of gray…

…scale try keeping just he last channel in that case
rapidsai · Apr 27, 2023 · 16263b0 · 16263b0
1 parent 2785f7d
commit 16263b0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/cucim/src/cucim/skimage/registration/tests/test_phase_cross_correlation.py b/python/cucim/src/cucim/skimage/registration/tests/test_phase_cross_correlation.py
@@ -180,6 +180,8 @@ def test_mismatch_offsets_size():
 )
 def test_disambiguate_2d(shift0, shift1):
     image = cp.array(eagle()[500:, 900:])  # use a highly textured image region
+    if image.ndim == 3:
+        image = image[..., 0]
     assert image.ndim == 2
     shift = (shift0, shift1)
     origin0 = []