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