def setUp(self):
self.node = Node(42)
class TestNode(unittest.TestCase):
def setUp(self):
self.node = Node(42)
def test_node_init(self):
self.assertEquals(self.node.get_id(), 42)
def test_node_parents(self):
parent_node = Node('parent')
self.node.add_parent(parent_node, 'edge')
count = 0
for parent in self.node.get_parents():
self.assertEquals(parent, parent_node)
count += 1
self.assertEquals(count, 1)
another_node = Node('another')
self.node.add_parent(parent_node, 'edge')
self.node.add_parent(another_node, 'edge')
for parent in self.node.get_parents():
self.assertIn(parent, (parent_node, another_node))
self.assertIn(parent.get_id(), ('parent', 'another'))
count += 1
self.assertEquals(count, 3)
def test_node_children(self):
child_node = Node('child')
self.node.add_child(child_node, 'edge')
count = 0
for child in self.node.get_children():
self.assertEquals(child, child_node)
count += 1
self.assertEquals(count, 1)
another_node = Node('another')
self.node.add_child(child_node, 'edge')
self.node.add_child(another_node, 'edge')
for child in self.node.get_children():
self.assertIn(child, (child_node, another_node))
self.assertIn(child.get_id(), ('child', 'another'))
count += 1
self.assertEquals(count, 3)
def test_node_edges(self):
parent_node = Node('parent')
child_node = Node('child')
self.node.add_parent(parent_node, 'parent_edge')
self.node.add_child(child_node, 'child_edge')
self.assertEquals(self.node.get_edge_to(child_node), 'child_edge')
self.assertEquals(self.node.get_edge_from(parent_node), 'parent_edge')