import time
import pidly
from common_pidly import setup_idl_environment
import scoop
from scoop import utils


def test_scoop(arg):
    worker_time = time.time()
    scoop.logger.info("Starting")
    scoop.logger.info(arg)
    idl = pidly.IDL()
    #setup_idl_environment(idl)
    idl('cpu,tpool_nthreads=1')
    idl('a = fft(dist(4096, 4096))')
    #time.sleep(1)
    #t = 1 / 0
    scoop.logger.info("Finished")
    scoop.logger.info('All done in {:.3f} seconds'.format(time.time() - worker_time))


if __name__ == '__main__':
    scoop_time = time.time()
    returnValues = list(scoop.futures.map_as_completed(test_scoop, range(12)))