Refactor TEST_CASE_RESBLOCK to use dict_product for cleaner test case generation

Signed-off-by: R. Garcia-Dias <[email protected]>

Author: garciadias

tests/networks/blocks/test_segresnet_block.py

@@ -18,23 +18,26 @@
 
 from monai.networks import eval_mode
 from monai.networks.blocks.segresnet_block import ResBlock
-
-TEST_CASE_RESBLOCK = []
-for spatial_dims in range(2, 4):
-    for in_channels in range(1, 4):
-        for kernel_size in [1, 3]:
-            for norm in [("group", {"num_groups": 1}), "batch", "instance"]:
-                test_case = [
-                    {
-                        "spatial_dims": spatial_dims,
-                        "in_channels": in_channels,
-                        "kernel_size": kernel_size,
-                        "norm": norm,
-                    },
-                    (2, in_channels, *([16] * spatial_dims)),
-                    (2, in_channels, *([16] * spatial_dims)),
-                ]
-                TEST_CASE_RESBLOCK.append(test_case)
+from tests.test_utils import dict_product
+
+TEST_CASE_RESBLOCK = [
+    [
+        {
+            "spatial_dims": params["spatial_dims"],
+            "in_channels": params["in_channels"],
+            "kernel_size": params["kernel_size"],
+            "norm": params["norm"],
+        },
+        (2, params["in_channels"], *([16] * params["spatial_dims"])),
+        (2, params["in_channels"], *([16] * params["spatial_dims"])),
+    ]
+    for params in dict_product(
+        spatial_dims=range(2, 4),
+        in_channels=range(1, 4),
+        kernel_size=[1, 3],
+        norm=[("group", {"num_groups": 1}), "batch", "instance"],
+    )
+]
 
 
 class TestResBlock(unittest.TestCase):