def test_adding_links_out_to_a_node(self): node = Node('nameOne') node_two = Node('nameTwo') node_three = Node('nameThree') node.add_outward_link(node_two) node.add_outward_link(node_three) links_out = node.links_out() self.assertTrue(node_two in links_out) self.assertTrue(node_three in links_out)
def test_build_graph(self): nodeA = Node('A') nodeA.add_outward_link('B') nodeA.add_outward_link('C') nodeB = Node('B') nodeB.add_outward_link('A') nodeB.add_outward_link('E') nodeC = Node('C') nodeC.add_outward_link('A') nodeC.add_outward_link('E') nodeC.add_outward_link('B') nodeD = Node('D') nodeD.add_outward_link('B') nodeE = Node('E') nodeE.add_outward_link('B') nodeE.add_outward_link('C') data = {} data['A'] = {'B', 'C'} data['B'] = {'A', 'E'} data['C'] = {'A', 'E', 'B'} data['D'] = {'B'} data['E'] = {'B', 'C'} nodes = build_graph(data) self.assertTrue(nodeA in nodes) self.assertTrue(nodeB in nodes) self.assertTrue(nodeC in nodes) self.assertTrue(nodeD in nodes) self.assertTrue(nodeE in nodes)