Fix RandGridPatch whole-dimension patch sizes

Signed-off-by: Rajioba1 <raji.lukmon@gmail.com>

Author: Rajioba1

monai/transforms/spatial/array.py

@@ -3567,7 +3567,7 @@ def randomize(self, array):
         else:
             min_offset = ensure_tuple_rep(self.min_offset, len(self.patch_size))
         if self.max_offset is None:
-            max_offset = tuple(s % p for s, p in zip(array.shape[1:], self.patch_size))
+            max_offset = tuple(s % p if p else 0 for s, p in zip(array.shape[1:], self.patch_size))
         else:
             max_offset = ensure_tuple_rep(self.max_offset, len(self.patch_size))
 

tests/transforms/spatial/test_rand_grid_patch.py

@@ -66,6 +66,8 @@
     A,
     [A[:, 1:3, 1:3]],
 ]
+TEST_CASE_14 = [{"patch_size": (0, 2)}, A, [A[:, :, :2], A[:, :, 2:]]]
+TEST_CASE_15 = [{"patch_size": (None, 2)}, A, [A[:, :, :2], A[:, :, 2:]]]
 
 TEST_CASE_META_0 = [
     {"patch_size": (2, 2)},
@@ -102,6 +104,8 @@
     TEST_SINGLE.append([p, *TEST_CASE_11])
     TEST_SINGLE.append([p, *TEST_CASE_12])
     TEST_SINGLE.append([p, *TEST_CASE_13])
+    TEST_SINGLE.append([p, *TEST_CASE_14])
+    TEST_SINGLE.append([p, *TEST_CASE_15])
 
 
 class TestRandGridPatch(unittest.TestCase):

tests/transforms/spatial/test_rand_grid_patchd.py

@@ -65,6 +65,8 @@
     {"image": A},
     [A[:, 1:3, 1:3]],
 ]
+TEST_CASE_14 = [{"patch_size": (0, 2)}, {"image": A}, [A[:, :, :2], A[:, :, 2:]]]
+TEST_CASE_15 = [{"patch_size": (None, 2)}, {"image": A}, [A[:, :, :2], A[:, :, 2:]]]
 
 TEST_SINGLE = []
 for p in TEST_NDARRAYS:
@@ -82,6 +84,8 @@
     TEST_SINGLE.append([p, *TEST_CASE_11])
     TEST_SINGLE.append([p, *TEST_CASE_12])
     TEST_SINGLE.append([p, *TEST_CASE_13])
+    TEST_SINGLE.append([p, *TEST_CASE_14])
+    TEST_SINGLE.append([p, *TEST_CASE_15])
 
 
 class TestRandGridPatchd(unittest.TestCase):