Skip to content

Commit 5cf2684

Browse files
committed
Fix merge error
1 parent 96849d1 commit 5cf2684

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

Lib/test/_test_multiprocessing.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2581,10 +2581,18 @@ def test_async(self):
25812581
self.assertTimingAlmostEqual(get.elapsed, TIMEOUT1)
25822582

25832583
def test_async_timeout(self):
2584-
res = self.pool.apply_async(sqr, (6, 5 * TIMEOUT2))
2585-
get = TimingWrapper(res.get)
2586-
self.assertRaises(multiprocessing.TimeoutError, get, timeout=TIMEOUT2)
2587-
self.assertTimingAlmostEqual(get.elapsed, TIMEOUT2)
2584+
p = self.Pool(3)
2585+
try:
2586+
event = threading.Event() if self.TYPE == 'threads' else None
2587+
res = p.apply_async(sqr, (6, TIMEOUT2 + support.SHORT_TIMEOUT, event))
2588+
get = TimingWrapper(res.get)
2589+
self.assertRaises(multiprocessing.TimeoutError, get, timeout=TIMEOUT2)
2590+
self.assertTimingAlmostEqual(get.elapsed, TIMEOUT2)
2591+
finally:
2592+
if event is not None:
2593+
event.set()
2594+
p.terminate()
2595+
p.join()
25882596

25892597
def test_imap(self):
25902598
it = self.pool.imap(sqr, list(range(10)))

0 commit comments

Comments
 (0)