def test_remove_complex_all(self): insert(self.subtree, parent=find('A')) Menu.remove(find('A'), all=True) expected = ((x,) for x in 'B,C,D,E,F'.split(',')) self.assertEqualTree(expected)
def test_remove_complex_node_only(self): insert(self.subtree, parent=find('A')) Menu.remove(find('A')) expected = ((x,) for x in 'a,b,c,d,e,f,B,C,D,E,F'.split(',')) self.assertEqualTree(expected)
def test_remove_simple(self): Menu.remove(find('A')) Menu.remove(find('F'), all=True) expected = ((x,) for x in 'B,C,D,E'.split(',')) self.assertEqualTree(expected)