Fix test mistakes

Signed-off-by: R. Garcia-Dias <rafaelagd@gmail.com>

Author: garciadias

tests/networks/blocks/test_patchembedding.py

@@ -45,16 +45,16 @@
     )
 ]
 
+img_size = 96
 TEST_CASE_PATCHEMBED = [
     [
         params,
-        (2, params["in_chans"], *([params["img_size"]] * params["spatial_dims"])),
-        (2, (params["img_size"] // params["patch_size"]) ** params["spatial_dims"], params["embed_dim"]),
+        (2, params["in_chans"], *([img_size] * params["spatial_dims"])),
+        (2, params["embed_dim"], *([img_size // params["patch_size"]]) * params["spatial_dims"]),
     ]
     for params in dict_product(
         patch_size=[2],
         in_chans=[1, 4],
-        img_size=[96],
         embed_dim=[6, 12],
         norm_layer=[nn.LayerNorm],
         spatial_dims=[2, 3],

tests/transforms/test_gibbs_noise.py

@@ -25,9 +25,10 @@
 
 _, has_torch_fft = optional_import("torch.fft", name="fftshift")
 
-params = {"shape": ((128, 64), (64, 48, 80)), "input_type": TEST_NDARRAYS if has_torch_fft else [np.array]}
-TEST_CASES = list(dict_product(format="list", **params))
-
+TEST_CASES = dict_product(shape=((128, 64), (64, 48, 80)), input_type=TEST_NDARRAYS if has_torch_fft else [np.array])
+TEST_CASES = [
+    [p_dict["shape"], p_dict["input_type"]] for p_dict in TEST_CASES
+]
 
 class TestGibbsNoise(unittest.TestCase):
     def setUp(self):