def test_remove_duplicate(self):
n = TreeNode(None)
ch = 'a'
n.add_child(ch)
ch_num = len(n.children)
n.remove_child(ch)
self.assertEquals(ch_num - 1, len(n.children))
n.remove_child(ch)
self.assertEquals(ch_num - 1, len(n.children))
def test_remove_child(self):
n = TreeNode(None)
ch = 'a'
n.add_child(ch)
self.assertTrue(ch in n.children)
self.assertTrue(TreeNode(ch) in n.children)
ch_num = len(n.children)
n.remove_child(ch)
self.assertEquals(ch_num - 1, len(n.children))
self.assertTrue(ch not in n.children)
self.assertTrue(TreeNode(ch) not in n.children)
ch = 'b'
n.add_child(ch)
ch_num = len(n.children)
n.remove_child(TreeNode(ch))
self.assertEquals(ch_num - 1, len(n.children))
