def some_function(n):
    for i in range(n):
        time.sleep(0.1)
        yield i

from parallel_ex import ParallelEx

def test_run_n_times():
    fun = partial(some_function, 10)
    ParallelEx.run_n_times(fun, n=5, total=10, n_cpu=2)
    # Total specifies the number of expected results.

def test_run_kwargs():
    totals = [25,8,6,22,12,6,10,9]
    kwargs = [{'n':i} for i in totals]
    ParallelEx.run_kwargs(some_function, kwargs, totals=totals, n_cpu=3)
    # Totals can specify different number of expected results for each execution.