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')