def test_server_cycle(self): c = make_system_composition() w = ScalaBench('jython') f = DacapoHarness() s = Send() r = Receive() p = Print() m = MockPipelineElement() m.inputs = Types(('a', int)) m.outputs = Types(('a', int)) c.jvm.workload = w c.flow = [w >> f >> s] j = Job(c, [r >> m >> m >> p]) self.assertFalse(j.check())
class PipelineElementHookTest(unittest.TestCase): def setUp(self): self.e = MockPipelineElement() self.list_ = [23, 42, 5] def test_pre_hooks(self): self.e.hooks = [ Hook(lambda: self.list_.__setitem__(0, 1)), Hook(lambda: self.list_.__setitem__(1, 1)), Hook(lambda: self.list_.__setitem__(2, 1))] self.e.run() self.assertListEqual(self.list_, [1, 1, 1]) def test_post_hooks(self): self.e.hooks = [ Hook(teardown=lambda: self.list_.__setitem__(0, 1)), Hook(teardown=lambda: self.list_.__setitem__(1, 1)), Hook(teardown=lambda: self.list_.__setitem__(2, 1))] self.e.run() self.assertListEqual(self.list_, [1, 1, 1])
def setUp(self): self.e = MockPipelineElement() self.list_ = [23, 42, 5]