def test_usage_with_different_functions(self):
def f(x):
return x+1
def f2(x):
return x+2
wq = WorkerQueue()
fr = FuncResult(f)
fr2 = FuncResult(f2)
wq.do(fr, 1)
wq.do(fr2, 1)
wq.wait()
wq.stop()
self.assert_(fr.result == 2)
self.assertEqual(fr2.result, 3)
def test_FuncResult(self):
"""Ensure FuncResult sets its result and exception attributes"""
# Results are stored in result attribute
fr = FuncResult(lambda x: x + 1)
fr(2)
self.assertEqual(fr.result, 3)
# Exceptions are store in exception attribute
self.assertIsNone(fr.exception, "no exception should be raised")
exception = ValueError("rast")
def x(sdf):
raise exception
fr = FuncResult(x)
fr(None)
self.assertIs(fr.exception, exception)