test_map_parallel.py: refactor using pytest.mark

tests/test_map_parallel.py

@@ -1,4 +1,6 @@
-from itertools import starmap
+from itertools import product
+
+from pytest import mark
 
 from map_parallel import map_parallel
 from map_parallel import starmap_parallel
@@ -8,25 +10,30 @@
     [5, 12, 13],
     [1, 2, 3]
 ]
+args = list(map(list, zip(*ARGS)))
+
+# MPI is difficult to test
+cases = list(product(('multiprocessing', 'multithreading'), (None, 1, 2)))
 
 
 def f(x, y, z):
     return x * x + y * y == z * z
 
 
-def test_map_parallel():
-    args = list(map(list, zip(*ARGS)))
-    truth = list(map(f, *args))
-    # MPI is difficult to test
-    for mode in ('multiprocessing', 'multithreading'):
-        for processes in (None, 1, 2):
-            assert map_parallel(f, *args, mode=mode, processes=processes) == truth
+truth = list(map(f, *args))
+
+
+@mark.parametrize(
+    'mode, processes',
+    cases,
+)
+def test_map_parallel(mode, processes):
+    assert map_parallel(f, *args, mode=mode, processes=processes) == truth
 
 
-def test_starmap_parallel():
-    args = ARGS
-    truth = list(starmap(f, args))
-    # MPI is difficult to test
-    for mode in ('multiprocessing', 'multithreading'):
-        for processes in (None, 1, 2):
-            assert starmap_parallel(f, args, mode=mode, processes=processes) == truth
+@mark.parametrize(
+    'mode, processes',
+    cases,
+)
+def test_starmap_parallel(mode, processes):
+    assert starmap_parallel(f, ARGS, mode=mode, processes=processes) == truth