Skip to content

Commit c000178

Browse files
committed
Add None patch size regression tests
Signed-off-by: Rajioba1 <196587943+Rajioba1@users.noreply.github.com>
1 parent 2ce2116 commit c000178

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

tests/transforms/spatial/test_rand_grid_patch.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from monai.data import MetaTensor, set_track_meta
2121
from monai.transforms.spatial.array import RandGridPatch
2222
from monai.utils import set_determinism
23-
from tests.test_utils import TEST_NDARRAYS, assert_allclose
23+
from tests.test_utils import TEST_NDARRAYS, TEST_NDARRAYS_NO_META_TENSOR, assert_allclose
2424

2525
A = np.arange(16).repeat(3).reshape(4, 4, 3).transpose(2, 0, 1)
2626
A11 = A[:, :2, :2]
@@ -67,6 +67,7 @@
6767
[A[:, 1:3, 1:3]],
6868
]
6969
TEST_CASE_14 = [{"patch_size": (0, 2)}, A, [A[:, :, :2], A[:, :, 2:]]]
70+
TEST_CASE_15 = [{"patch_size": (None, 2)}, A, [A[:, :, :2], A[:, :, 2:]]]
7071

7172
TEST_CASE_META_0 = [
7273
{"patch_size": (2, 2)},
@@ -104,6 +105,8 @@
104105
TEST_SINGLE.append([p, *TEST_CASE_12])
105106
TEST_SINGLE.append([p, *TEST_CASE_13])
106107
TEST_SINGLE.append([p, *TEST_CASE_14])
108+
for p in TEST_NDARRAYS_NO_META_TENSOR:
109+
TEST_SINGLE.append([p, *TEST_CASE_15])
107110

108111

109112
class TestRandGridPatch(unittest.TestCase):

tests/transforms/spatial/test_rand_grid_patchd.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
from monai.transforms.spatial.dictionary import RandGridPatchd
2121
from monai.utils import set_determinism
22-
from tests.test_utils import TEST_NDARRAYS, assert_allclose
22+
from tests.test_utils import TEST_NDARRAYS, TEST_NDARRAYS_NO_META_TENSOR, assert_allclose
2323

2424
A = np.arange(16).repeat(3).reshape(4, 4, 3).transpose(2, 0, 1)
2525
A11 = A[:, :2, :2]
@@ -66,6 +66,7 @@
6666
[A[:, 1:3, 1:3]],
6767
]
6868
TEST_CASE_14 = [{"patch_size": (0, 2)}, {"image": A}, [A[:, :, :2], A[:, :, 2:]]]
69+
TEST_CASE_15 = [{"patch_size": (None, 2)}, {"image": A}, [A[:, :, :2], A[:, :, 2:]]]
6970

7071
TEST_SINGLE = []
7172
for p in TEST_NDARRAYS:
@@ -84,6 +85,8 @@
8485
TEST_SINGLE.append([p, *TEST_CASE_12])
8586
TEST_SINGLE.append([p, *TEST_CASE_13])
8687
TEST_SINGLE.append([p, *TEST_CASE_14])
88+
for p in TEST_NDARRAYS_NO_META_TENSOR:
89+
TEST_SINGLE.append([p, *TEST_CASE_15])
8790

8891

8992
class TestRandGridPatchd(unittest.TestCase):

0 commit comments

Comments
 (0)