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)
