def test_transform_raise(self):
future = BatchAsyncResult()
future.set_exception(ValueError())
tfuture = transform(future, lambda: "NOT CALLED")
gevent.sleep(0)
self.assertTrue(tfuture.ready())
self.assertFalse(tfuture.successful())
self.assertRaises(ValueError, tfuture.get)
def test_transform(self):
def multiply_by(value_f, m=3):
return value_f.get() * m
future = BatchAsyncResult()
def finish_it():
future.set(2)
tfuture = transform(future, multiply_by, m=4)
self._do_test_future(tfuture, finish_it)
self.assertEquals(8, tfuture.get())
