def test_run_dataflow_single_only(self):
# at timestamp = 0
config = {ContextAggregator.PM:ContextAggregator.SINGLE_ONLY_MODE}
c0 = ContextAggregator(id=0, config=config)
c1 = ContextAggregator(id=1, config=config)
c2 = ContextAggregator(id=2, config=config)
# First round
## compute
c0.process_to_set_output(neighbors=[1], timestamp = 0)
c1.process_to_set_output(neighbors=[0,2], timestamp = 0)
c2.process_to_set_output(neighbors=[1], timestamp = 0)
## communication
r0_1 = c0.send(neighbor=1, timestamp=0)
r1_0 = c1.send(neighbor=0, timestamp=0)
r1_2 = c1.send(neighbor=2, timestamp=0)
r2_1 = c2.send(neighbor=1, timestamp=0)
c1.receive(from_node=0, contexts=r0_1[1], timestamp=0)
c0.receive(from_node=1, contexts=r1_0[0], timestamp=0)
c2.receive(from_node=1, contexts=r1_2[2], timestamp=0)
c1.receive(from_node=2, contexts=r2_1[1], timestamp=0)
r1_0 = c1.get_received_data(from_node=0)
r0_1 = c0.get_received_data(from_node=1)
r1_2 = c1.get_received_data(from_node=2)
r2_1 = c2.get_received_data(from_node=1)
self.assertTrue(contexts_to_standard(r1_0), [[0],[]])
self.assertTrue(contexts_to_standard(r0_1), [[1],[]])
self.assertTrue(contexts_to_standard(r1_2), [[2],[]])
self.assertTrue(contexts_to_standard(r2_1), [[1],[]])
# Second round
## compute
o0 = c0.process_to_set_output(neighbors=[1], timestamp = 0)
o1 = c1.process_to_set_output(neighbors=[0,2], timestamp = 0)
o2 = c2.process_to_set_output(neighbors=[1], timestamp = 0)
self.assertTrue(contexts_to_standard(c0.get_singles(timestamp=0)) == [[0,1],[]])
self.assertTrue(contexts_to_standard(c1.get_singles(timestamp=0)) == [[0,1,2],[]])
self.assertTrue(contexts_to_standard(c2.get_singles(timestamp=0)) == [[1,2],[]])
self.assertTrue(o0 == {1: [[], []]})
self.assertTrue(o1 == {0: [[2], []], 2:[[0], []]})
self.assertTrue(o2 == {1: [[], []]})
## communication
self.assertTrue(c0.is_nothing_to_send())
self.assertTrue(c2.is_nothing_to_send())
r1_0 = c1.send(neighbor=0, timestamp=0)
r1_2 = c1.send(neighbor=2, timestamp=0)
self.assertTrue(contexts_to_standard(r1_0[0]) == [[2],[]])
self.assertTrue(contexts_to_standard(r1_2[2]) == [[0],[]])
c0.receive(from_node=1, contexts=r1_0[0], timestamp=0)
c2.receive(from_node=1, contexts=r1_2[2], timestamp=0)
r0_1 = c0.get_received_data(from_node=1)
r2_1 = c2.get_received_data(from_node=1)
self.assertTrue(contexts_to_standard(r0_1) == [[2],[]])
self.assertTrue(contexts_to_standard(r2_1) == [[0],[]])
# Third iteration
o0 = c0.process_to_set_output(neighbors=[1], timestamp = 0)
o1 = c1.process_to_set_output(neighbors=[0,2], timestamp = 0)
o2 = c2.process_to_set_output(neighbors=[1], timestamp = 0)
self.assertTrue(contexts_to_standard(c0.get_singles(timestamp=0)) == [[0,1,2],[]])
self.assertTrue(contexts_to_standard(c1.get_singles(timestamp=0)) == [[0,1,2],[]])
self.assertTrue(contexts_to_standard(c2.get_singles(timestamp=0)) == [[0,1,2],[]])
self.assertTrue(o0 == {1: [[], []]})
#self.assertTrue(o1 == {0: [[], []], 2:[[], []]})
self.assertTrue(o1 == {}) # {0: [[], []], 2:[[], []]})
self.assertTrue(o2 == {1: [[], []]})
self.assertTrue(c0.is_nothing_to_send())
self.assertTrue(c1.is_nothing_to_send())
self.assertTrue(c2.is_nothing_to_send())
