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)