def test_create_node(self):
"A node can be created with children"
style = Style()
child1 = Node(style=style)
child2 = Node(style=style)
child3 = Node(style=style)
node = Node(style=style, children=[child1, child2, child3])
self.assertEqual(node.children, [child1, child2, child3])
self.assertTrue(node.can_have_children)
# The node is the root as well.
self.assertIsNone(node.parent)
self.assertEqual(node.root, node)
# The children all point at the node.
self.assertEqual(child1.parent, node)
self.assertEqual(child1.root, node)
self.assertEqual(child2.parent, node)
self.assertEqual(child2.root, node)
self.assertEqual(child3.parent, node)
self.assertEqual(child3.root, node)
# Create another node
new_node = Node(style=style, children=[])
self.assertEqual(new_node.children, [])
self.assertTrue(new_node.can_have_children)
# Add the old node as a child of the new one.
new_node.add(node)
# The new node is the root
self.assertIsNone(new_node.parent)
self.assertEqual(new_node.root, new_node)
# The node is the root as well.
self.assertEqual(node.parent, new_node)
self.assertEqual(node.root, new_node)
# The children all point at the node.
self.assertEqual(child1.parent, node)
self.assertEqual(child1.root, new_node)
self.assertEqual(child2.parent, node)
self.assertEqual(child2.root, new_node)
self.assertEqual(child3.parent, node)
self.assertEqual(child3.root, new_node)
def test_add(self):
"Nodes can be added as children to another node"
style = Style()
node = Node(style=style, children=[])
child = Node(style=style)
node.add(child)
self.assertTrue(child in node.children)
self.assertEqual(child.parent, node)
self.assertEqual(child.root, node.root)
def test_create_leaf(self):
"A leaf can be created"
style = Style()
leaf = Node(style=style)
self.assertIsNone(leaf._children)
self.assertEqual(leaf.children, [])
self.assertFalse(leaf.can_have_children)
# An unattached leaf is a root
self.assertIsNone(leaf.parent)
self.assertEqual(leaf.root, leaf)
# A leaf can't have children
child = Node(style=style)
with self.assertRaises(ValueError):
leaf.add(child)