def test_santoro2007(self):
N_ITERS = 5
N_NETWORKS = 15
N_NODES_STEP = 5
node_range = 100
nets = [
[(100, 100)],
[(100, 100), (175, 250), (250, 175), (100, 250),
(175, 175), (100, 175)],
[(100, 100), (150, 200), (175, 175), (175, 100),
(250, 175), (250, 250), (325, 250), (325, 325), (325, 400)]
]
for i, node_positions in enumerate(nets, start=1):
net = Network()
for node_pos in node_positions:
net.add_node(pos=node_pos, commRange=node_range)
name = 'Special %d' % i
net.algorithms = (YoYo, )
sim = Simulation(net, logLevel='WARNING')
sim.run()
min_id = min(sim.network.nodes(), key=lambda node: node.id).id
for node in sim.network.nodes():
if node.id == min_id:
# Check if the node with the smallest ID is the LEADER
assert node.status == 'LEADER', \
'%s: Node %d has status %s, not LEADER' % \
(name, node.id, node.status)
else:
# Check if every other node is PRUNED
assert node.status == 'PRUNED', \
'%s: Node %d has status %s, not PRUNED' % \
(name, node.id, node.status)
for i in range(N_ITERS):
for n_nodes in range(N_NODES_STEP,
N_NETWORKS*N_NODES_STEP+N_NODES_STEP,
N_NODES_STEP):
net_gen = NetworkGenerator(n_nodes)
net = net_gen.generate_random_network()
name = 'Random %d, %d nodes' % (i, n_nodes)
net.algorithms = (YoYo, )
sim = Simulation(net, logLevel='WARNING')
sim.run()
min_id = min(sim.network.nodes(), key=lambda node: node.id).id
for node in sim.network.nodes():
if node.id == min_id:
# Check if the node with the smallest ID is the LEADER
assert node.status == 'LEADER', \
'%s: Node %d has status %s, not LEADER' % \
(name, node.id, node.status)
else:
# Check if every other node is PRUNED
assert node.status == 'PRUNED', \
'%s: Node %d has status %s, not PRUNED' % \
(name, node.id, node.status)
