def test_is_marked(self):
#
# Mark new node (generic)
#
node = Tree(n_children=4)
node.mark()
self.assertTrue(node.is_marked(), 'Node should be marked.')
#
# Unmark node
#
node.unmark()
self.assertFalse(node.is_marked(), 'Node should not be marked.')
#
# Mark node (specific)
#
node.mark(1)
self.assertTrue(node.is_marked(), 'Node should be marked.')
self.assertTrue(node.is_marked(1), 'Node should be marked 1.')
self.assertFalse(node.is_marked(2), 'Node should not be marked 2.')
def test_mark(self):
node = Tree(n_children=3)
node.mark()
self.assertTrue(node.is_marked(), 'Node should be marked.')
