def terminate(self):

util.debug('terminating pool')

self._state = TERMINATE

self._worker_handler._state = TERMINATE

self._change_notifier.put(None)

self._terminate()

def join(self):

# this is guaranteed to only be called once

util.debug('finalizing pool')

# Notify that the worker_handler state has been changed so the

# _handle_workers loop can be unblocked (and exited) in order to

# send the finalization sentinel all the workers.

worker_handler._state = TERMINATE

change_notifier.put(None)

task_handler._state = TERMINATE

util.debug('helping task handler/workers to finish')

for (j, res) in enumerate(results):

self.assertEqual(res.get(), sqr(j))

def test_worker_finalization_via_atexit_handler_of_multiprocessing(self):

# tests cases against bpo-38744 and bpo-39360

cmd = '''if 1:

from multiprocessing import Pool

problem = None

class A:

def __init__(self):

self.pool = Pool(processes=1)

def test():

global problem

problem = A()

problem.pool.map(float, tuple(range(10)))

if __name__ == "__main__":

test()

'''

rc, out, err = test.support.script_helper.assert_python_ok('-c', cmd)

self.assertEqual(rc, 0)

#

# Test of creating a customized manager class

#

Ensure all workers exit when finalizing a :class:`multiprocessing.Pool` implicitly via the module finalization

handlers of multiprocessing. This fixes a deadlock situation that can be experienced when the Pool is not

properly finalized via the context manager or a call to ``multiprocessing.Pool.terminate``. Patch by Batuhan Taskaya

and Pablo Galindo.