def test_output(self):
from sys import stdout
if not hasattr(stdout, 'getvalue'):
self.skipTest("cannot grab stdout")
par, func = parallel_loop(f, n_jobs=None, verbose=10)
self.assertEqual(stdout.getvalue().strip().split(' ')[-1], 'serially')
par, func = parallel_loop(f, n_jobs=2, verbose=10)
self.assertEqual(stdout.getvalue().strip().split(' ')[-1], 'parallel')
def test_parallel_loop(self):
verbose = 0
# reference list comprehension
ref = [f(i) for i in range(10)]
reg = [g(i, j, 5) for i, j in enumerate(range(10, 20))]
reh = [h(i) for i in range(10)]
# test non-parallel execution
par, func = parallel_loop(f, n_jobs=None, verbose=verbose)
self.assertEqual(ref, par(func(i) for i in range(10)))
# multiple arguments
par, func = parallel_loop(g, n_jobs=None, verbose=verbose)
self.assertEqual(reg, par(func(i, j, 5) for i, j in enumerate(range(10, 20))))
# multiple return values
par, func = parallel_loop(h, n_jobs=None, verbose=verbose)
self.assertEqual(reh, par(func(i) for i in range(10)))
# test non-parallel execution with joblib
par, func = parallel_loop(f, n_jobs=1, verbose=verbose)
b = par(func(i) for i in range(10))
self.assertEqual(ref, par(func(i) for i in range(10)))
# multiple arguments
par, func = parallel_loop(g, n_jobs=1, verbose=verbose)
self.assertEqual(reg, par(func(i, j, 5) for i, j in enumerate(range(10, 20))))
# multiple return values
par, func = parallel_loop(h, n_jobs=1, verbose=verbose)
self.assertEqual(reh, par(func(i) for i in range(10)))
# test parallel execution with joblib
par, func = parallel_loop(f, n_jobs=2, verbose=verbose)
b = par(func(i) for i in range(10))
self.assertEqual(ref, par(func(i) for i in range(10)))
# multiple arguments
par, func = parallel_loop(g, n_jobs=2, verbose=verbose)
self.assertEqual(reg, par(func(i, j, 5) for i, j in enumerate(range(10, 20))))
# multiple return values
par, func = parallel_loop(h, n_jobs=2, verbose=verbose)
self.assertEqual(reh, par(func(i) for i in range(10)))
# test parallel execution with joblib
par, func = parallel_loop(f, n_jobs=-1, verbose=verbose)
b = par(func(i) for i in range(10))
self.assertEqual(ref, par(func(i) for i in range(10)))
# multiple arguments
par, func = parallel_loop(g, n_jobs=-1, verbose=verbose)
self.assertEqual(reg, par(func(i, j, 5) for i, j in enumerate(range(10, 20))))
# multiple return values
par, func = parallel_loop(h, n_jobs=-1, verbose=verbose)
self.assertEqual(reh, par(func(i) for i in range(10)))
# test parallel execution with joblib
par, func = parallel_loop(f, n_jobs=10, verbose=verbose)
b = par(func(i) for i in range(10))
self.assertEqual(ref, par(func(i) for i in range(10)))
# multiple arguments
par, func = parallel_loop(g, n_jobs=10, verbose=verbose)
self.assertEqual(reg, par(func(i, j, 5) for i, j in enumerate(range(10, 20))))
# multiple return values
par, func = parallel_loop(h, n_jobs=10, verbose=verbose)
self.assertEqual(reh, par(func(i) for i in range(10)))
