Fix multiprocessing.event to match the new threading.Event API

This commit is contained in:
Jesse Noller 2009-04-01 03:45:50 +00:00
parent a83da3507f
commit 02cb0eb231
3 changed files with 20 additions and 7 deletions

View File

@ -836,6 +836,12 @@ object -- see :ref:`multiprocessing-managers`.
.. class:: Event() .. class:: Event()
A clone of :class:`threading.Event`. A clone of :class:`threading.Event`.
This method returns the state of the internal semaphore on exit, so it
will always return ``True`` except if a timeout is given and the operation
times out.
.. versionchanged:: 2.7
Previously, the method always returned ``None``.
.. class:: Lock() .. class:: Lock()

View File

@ -301,5 +301,10 @@ def wait(self, timeout=None):
self._flag.release() self._flag.release()
else: else:
self._cond.wait(timeout) self._cond.wait(timeout)
if self._flag.acquire(False):
self._flag.release()
return True
return False
finally: finally:
self._cond.release() self._cond.release()

View File

@ -749,20 +749,22 @@ def test_event(self):
# Removed temporaily, due to API shear, this does not # Removed temporaily, due to API shear, this does not
# work with threading._Event objects. is_set == isSet # work with threading._Event objects. is_set == isSet
#self.assertEqual(event.is_set(), False) self.assertEqual(event.is_set(), False)
self.assertEqual(wait(0.0), None) # Removed, threading.Event.wait() will return the value of the __flag
# instead of None. API Shear with the semaphore backed mp.Event
self.assertEqual(wait(0.0), False)
self.assertTimingAlmostEqual(wait.elapsed, 0.0) self.assertTimingAlmostEqual(wait.elapsed, 0.0)
self.assertEqual(wait(TIMEOUT1), None) self.assertEqual(wait(TIMEOUT1), False)
self.assertTimingAlmostEqual(wait.elapsed, TIMEOUT1) self.assertTimingAlmostEqual(wait.elapsed, TIMEOUT1)
event.set() event.set()
# See note above on the API differences # See note above on the API differences
# self.assertEqual(event.is_set(), True) self.assertEqual(event.is_set(), True)
self.assertEqual(wait(), None) self.assertEqual(wait(), True)
self.assertTimingAlmostEqual(wait.elapsed, 0.0) self.assertTimingAlmostEqual(wait.elapsed, 0.0)
self.assertEqual(wait(TIMEOUT1), None) self.assertEqual(wait(TIMEOUT1), True)
self.assertTimingAlmostEqual(wait.elapsed, 0.0) self.assertTimingAlmostEqual(wait.elapsed, 0.0)
# self.assertEqual(event.is_set(), True) # self.assertEqual(event.is_set(), True)
@ -771,7 +773,7 @@ def test_event(self):
#self.assertEqual(event.is_set(), False) #self.assertEqual(event.is_set(), False)
self.Process(target=self._test_event, args=(event,)).start() self.Process(target=self._test_event, args=(event,)).start()
self.assertEqual(wait(), None) self.assertEqual(wait(), True)
# #
# #