class RunServerPipelineTest(unittest.TestCase): def setUp(self): self.receive = Receive() self.j = Job([], [Edge(self.receive, MockPipelineElement())]) self.data = {'a': 1, 'b' : 2} self.j.receive = lambda: self.data def test_receive(self): self.assertDictEqual(self.receive.out, {}) self.j.run_server_pipeline() self.assertDictEqual(self.receive.out, {'results' : self.data}) def test_no_receivers(self): j = Job([], [DacapoHarness() >> Print()]) with self.assertRaises(ValueError): j.run_server_pipeline() def test_empty_flow(self): j = Job([], []) self.assertEqual(j.run_server_pipeline(), None)
def test_empty_flow(self): j = Job([], []) self.assertEqual(j.run_server_pipeline(), None)
def test_no_receivers(self): j = Job([], [DacapoHarness() >> Print()]) with self.assertRaises(ValueError): j.run_server_pipeline()