Kaydet (Commit) 02cb0eb2 authored tarafından Jesse Noller's avatar Jesse Noller

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

üst a83da350
...@@ -836,6 +836,12 @@ object -- see :ref:`multiprocessing-managers`. ...@@ -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()
......
...@@ -301,5 +301,10 @@ class Event(object): ...@@ -301,5 +301,10 @@ class Event(object):
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()
...@@ -749,20 +749,22 @@ class _TestEvent(BaseTestCase): ...@@ -749,20 +749,22 @@ class _TestEvent(BaseTestCase):
# 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 @@ class _TestEvent(BaseTestCase): ...@@ -771,7 +773,7 @@ class _TestEvent(BaseTestCase):
#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)
# #
# #
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment