Multiprocessing Freeze Support Test

multiprocessing_freeze_support.py

import sys
print('Arguments were', repr(sys.argv))

from argparse import ArgumentParser
import multiprocessing
from random import randrange
from time import sleep

def talk(x):
    sleep(x)
    print('Slept for {} seconds'.format(x))


def main():
    n = 10
    arg_list = [randrange(n) for x in range(n)]
    with multiprocessing.Pool() as pool:
        all_results = pool.imap(talk, arg_list)
        for result in all_results:
            pass

if __name__ == '__main__':
    multiprocessing.freeze_support()

    argument_parser = ArgumentParser()
    args = argument_parser.parse_args()

    # multiprocessing.set_start_method('spawn')
    multiprocessing.set_start_method('forkserver')
    main()