jobmanager/examples/pool_map.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import division, print_function

from os.path import split, dirname, abspath
import sys

import time
import numpy as np
import multiprocessing as mp

# Add parent directory to beginning of path variable
sys.path.insert(0, dirname(dirname(abspath(__file__))))

import jobmanager

def func(x, y, z):
    """Example function with only one argument"""
    time.sleep(x/10)
    return np.sum([x, y, z])

def wrapper(data):
    return func(*data)

# Create list of parameters
a = list()
for i in range(10):
    a.append([i, 2.34, 9])

# mp.Pool example:
p_mp = mp.Pool()
res_mp = p_mp.map(wrapper, a)

# equivalent to mp.Pool() but with progress bar:
p_jm = jobmanager.decorators.Pool()
res_jm = p_jm.map(wrapper, a)

assert res_mp == res_jm
print("result: ", res_jm)