def test_exceptions(self):
worker_pool = Pool(3, worker_names='WorkerThread')
def raise_on_1(foo):
if foo == 1:
raise TypeError('%s Boom!' % foo)
return foo
answers = worker_pool.imap_unordered(raise_on_1, xrange(3))
try:
[i for i in answers]
except TypeError, te:
self.assertEqual(str(te), '1 Boom!')
def test_exceptions(self):
worker_pool = Pool(3, worker_names='WorkerThread')
def raise_on_1(foo):
if foo == 1:
raise TypeError('%s Boom!' % foo)
return foo
answers = worker_pool.imap_unordered(raise_on_1, xrange(3))
try:
[i for i in answers]
except TypeError, te:
self.assertEqual(str(te), '1 Boom!')
# pylint: disable=E1101
self.assertIn("raise TypeError('%s Boom!' % foo)", te.original_traceback_string)