Add test for running function which takes no parameters with Scheduler

Author: TheGreatCabbage

test/test_scheduler.py

@@ -27,7 +27,14 @@
 
 from scheduler.Scheduler import Scheduler
 from scheduler.Task import Task
-from test.utils import _long_task, _get_input_output, assert_results, _func, _funcq
+from test.utils import (
+    _long_task,
+    _get_input_output,
+    assert_results,
+    _func,
+    _funcq,
+    _func_no_params,
+)
 
 
 def test_add_process():
@@ -61,6 +68,23 @@ def test_add():
     assert scheduler.finished
 
 
+def test_run_no_params():
+    """Tests whether `run()` works correctly on a function with no parameters."""
+    scheduler = Scheduler()
+
+    expected = _func_no_params()
+    for a in range(50):
+        scheduler.add(target=_func_no_params)
+
+    loop = asyncio.get_event_loop()
+    results: List = loop.run_until_complete(scheduler.run())
+
+    for r in results:
+        assert r == expected
+
+    assert scheduler.finished
+
+
 def test_multiprocess():
     """Tests whether `add()` works correctly with `multiprocess` instead of `multiprocessing`."""
     scheduler = Scheduler()
@@ -97,15 +121,16 @@ def test_run_blocking():
 
 
 def test_run_async():
+    """Tests whether `run()` works correctly."""
     scheduler = Scheduler()
 
     args, expected = _get_input_output()
     for a in args:
         scheduler.add(target=_func, args=a)
 
     loop = asyncio.get_event_loop()
-
     results: List[Tuple[int, int, int]] = loop.run_until_complete(scheduler.run())
+
     assert_results(expected, results)
 
     assert scheduler.finished

test/utils.py

@@ -55,6 +55,10 @@ def assert_results(expected, results):
             assert expected[i] != results[i + 1]
 
 
+def _func_no_params() -> float:
+    return 3.1415
+
+
 def _func(x, y, z) -> Tuple:
     return x ** 2, y ** 3, z ** 4