def _build_0123_graph(self):
"""Graph for this TestCase is as follows.
0
↙ ↘
1 2
↙
3
"""
nodes = [
route.Node('0'),
route.Node('1'),
route.Node('2'),
route.Node('3'),
]
nodes[0].add_link(nodes[1], 0)
nodes[0].add_link(nodes[2], 0)
nodes[1].add_link(nodes[3], 0)
self.graph.add_nodes(*nodes)
return nodes
def test_distance_of_link_to_node(self):
node = route.Node('node')
distance = 5
self.root_node.add_link(node, distance)
self.assertEquals(distance, self.root_node.links[node])
def test_can_add_link_to_node(self):
node = route.Node('node')
self.root_node.add_link(node, 1)
self.assertIn(node, self.root_node.links)
def setUp(self):
self.root_node = route.Node('root_node')
def test_get_node_gives_back_correct_node(self):
node = route.Node('1')
self.graph.add_nodes(node)
self.assertEquals(node, self.graph['1'])
def test_adding_existing_node_to_graph_does_not_duplicate(self):
node = route.Node('0')
self.graph.add_nodes(node, node)
self.assertListEqual([node], self.graph.nodes)
def test_can_add_nodes_to_graph(self):
nodes = [route.Node('0'), route.Node('1')]
self.graph.add_nodes(*nodes)
self.assertListEqual(nodes, self.graph.nodes)