Parametrized tests for test_mean_absolute_error.py (#2625)

Author: kamalojasv181

tests/ignite/metrics/test_mean_absolute_error.py

@@ -17,44 +17,40 @@ def test_no_update():
         mae.compute()
 
 
-def test_compute():
+@pytest.fixture(params=[item for item in range(4)])
+def test_case(request):
+
+    return [
+        (torch.randint(0, 10, size=(100, 1)), torch.randint(0, 10, size=(100, 1)), 1),
+        (torch.randint(-10, 10, size=(100, 5)), torch.randint(-10, 10, size=(100, 5)), 1),
+        # updated batches
+        (torch.randint(0, 10, size=(100, 1)), torch.randint(0, 10, size=(100, 1)), 16),
+        (torch.randint(-20, 20, size=(100, 5)), torch.randint(-20, 20, size=(100, 5)), 16),
+    ][request.param]
+
+
+@pytest.mark.parametrize("n_times", range(5))
+def test_compute(n_times, test_case):
 
     mae = MeanAbsoluteError()
 
-    def _test(y_pred, y, batch_size):
-        mae.reset()
-        if batch_size > 1:
-            n_iters = y.shape[0] // batch_size + 1
-            for i in range(n_iters):
-                idx = i * batch_size
-                mae.update((y_pred[idx : idx + batch_size], y[idx : idx + batch_size]))
-        else:
-            mae.update((y_pred, y, batch_size))
-
-        np_y = y.numpy()
-        np_y_pred = y_pred.numpy()
-
-        np_res = (np.abs(np_y_pred - np_y)).sum() / np_y.shape[0]
-        assert isinstance(mae.compute(), float)
-        assert mae.compute() == np_res
-
-    def get_test_cases():
-
-        test_cases = [
-            (torch.randint(0, 10, size=(100, 1)), torch.randint(0, 10, size=(100, 1)), 1),
-            (torch.randint(-10, 10, size=(100, 5)), torch.randint(-10, 10, size=(100, 5)), 1),
-            # updated batches
-            (torch.randint(0, 10, size=(100, 1)), torch.randint(0, 10, size=(100, 1)), 16),
-            (torch.randint(-20, 20, size=(100, 5)), torch.randint(-20, 20, size=(100, 5)), 16),
-        ]
-
-        return test_cases
-
-    for _ in range(5):
-        # check multiple random inputs as random exact occurencies are rare
-        test_cases = get_test_cases()
-        for y_pred, y, batch_size in test_cases:
-            _test(y_pred, y, batch_size)
+    y_pred, y, batch_size = test_case
+
+    mae.reset()
+    if batch_size > 1:
+        n_iters = y.shape[0] // batch_size + 1
+        for i in range(n_iters):
+            idx = i * batch_size
+            mae.update((y_pred[idx : idx + batch_size], y[idx : idx + batch_size]))
+    else:
+        mae.update((y_pred, y, batch_size))
+
+    np_y = y.numpy()
+    np_y_pred = y_pred.numpy()
+
+    np_res = (np.abs(np_y_pred - np_y)).sum() / np_y.shape[0]
+    assert isinstance(mae.compute(), float)
+    assert mae.compute() == np_res
 
 
 def _test_distrib_integration(device):