Add tests for map functions

Author: TheGreatCabbage

test/test_scheduler.py

@@ -39,6 +39,7 @@
         _funcq,
         _func_no_params,
         _func_numpy,
+        _func_no_return,
     )
 except:
     from utils import (
@@ -51,6 +52,7 @@
         _funcq,
         _func_no_params,
         _func_numpy,
+        _func_no_return,
     )
 
 
@@ -183,6 +185,35 @@ def test_map():
     assert scheduler.finished
 
 
+def test_map_no_args():
+    """
+    Tests whether `map()` works correctly with no araguments.
+    """
+    scheduler = Scheduler()
+
+    loop = asyncio.get_event_loop()
+    results: List = loop.run_until_complete(scheduler.map(target=_func_no_params))
+
+    expected = _func_no_params()
+
+    assert all([r == expected for r in results])
+    assert scheduler.finished
+
+
+def test_map_no_return():
+    """
+    Tests whether 'map()' works correctly when the target function 
+    does not return any values.
+    """
+    scheduler = Scheduler()
+
+    loop = asyncio.get_event_loop()
+    results = loop.run_until_complete(scheduler.map(target=_func_no_return))
+
+    assert all([r is None for r in results])
+    assert scheduler.finished
+
+
 def test_map_blocking():
     """Tests whether `map_blocking()` works correctly."""
     scheduler = Scheduler()

test/utils.py

@@ -48,7 +48,7 @@ def _get_input_output() -> Tuple[List, List]:
 
 
 def _get_input_output_numpy() -> Tuple[List, List]:
-    args = [(i, 5000000 + i ** 2) for i in range(15)]
+    args = [(i, 500000 + i ** 2) for i in range(15)]
     expected_output = [(_func_numpy(*a)) for a in args]
     return args, expected_output
 
@@ -77,6 +77,10 @@ def _func_numpy(x, y) -> ndarray:
     return numpy.arange(x, y)
 
 
+def _func_no_return(x: int) -> None:
+    pass
+
+
 def _func_no_params() -> float:
     return 3.1415