def test_callback(self): self.callback_called = False self.py_event = PyEvent() def callback(event, status): self.callback_called = True self.py_event.set() event = UserEvent(ctx) queue = Queue(ctx, ctx.devices[0]) queue.enqueue_wait_for_events(event) event2 = queue.marker() event2.add_callback(callback) self.assertEqual(event.status, Event.SUBMITTED) self.assertEqual(event2.status, Event.QUEUED) self.assertFalse(self.callback_called) event.complete() self.assertEqual(event.status, Event.COMPLETE) event2.wait() self.assertEqual(event2.status, Event.COMPLETE) event_is_set = self.py_event.wait(2) self.assertTrue(event_is_set, 'timed out waiting for callback') self.assertTrue(self.callback_called)
def test_status(self): event = UserEvent(ctx) self.assertEqual(event.status, Event.SUBMITTED) event.complete() self.assertEqual(event.status, Event.COMPLETE)
def test_wait(self): event = UserEvent(ctx) queue = Queue(ctx, ctx.devices[0]) queue.enqueue_wait_for_events(event) event2 = queue.marker() self.assertEqual(event.status, Event.SUBMITTED) self.assertEqual(event2.status, Event.QUEUED) event.complete() self.assertEqual(event.status, Event.COMPLETE) event2.wait() self.assertEqual(event2.status, Event.COMPLETE)