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)
