Add a couple unit tests that should take long enough that heartbeats will actually happen.

Author: dan-blanchard

tests/test_gridmap.py

@@ -23,6 +23,7 @@
 from __future__ import print_function, unicode_literals
 
 import logging
+from time import sleep
 
 import gridmap
 from gridmap import Job, process_jobs, grid_map
@@ -37,6 +38,15 @@
 logger = logging.getLogger(__name__)
 logger.debug('Path to gridmap: %s', gridmap)
 
+
+def compute_factorial_slow(n):
+    sleep(100)
+    ret = 1
+    for i in range(n):
+        ret = ret * (i + 1)
+    return ret
+
+
 def compute_factorial(n):
     ret = 1
     for i in range(n):
@@ -51,6 +61,13 @@ def test_map():
     eq_(expected, outputs)
 
 
+def test_map_slow():
+    inputs = [1, 2, 4, 8, 16]
+    expected = list(map(compute_factorial_slow, inputs))
+    outputs = grid_map(compute_factorial_slow, inputs, quiet=False)
+    eq_(expected, outputs)
+
+
 def make_jobs(inputvec, function):
     # create empty job vector
     jobs = []
@@ -71,3 +88,11 @@ def test_process_jobs():
     function_jobs = make_jobs(inputs, compute_factorial)
     outputs = process_jobs(function_jobs, quiet=False)
     eq_(expected, outputs)
+
+
+def test_process_jobs_slow():
+    inputs = [1, 2, 4, 8, 16]
+    expected = list(map(compute_factorial_slow, inputs))
+    function_jobs = make_jobs(inputs, compute_factorial_slow)
+    outputs = process_jobs(function_jobs, quiet=False)
+    eq_(expected, outputs)