def test_node_add_multiple_children_then_delete_one(self): n = Node() left = Node("left") right = Node("right") n.add_child(left) n.add_child(right) n.remove_child("left") self.assertEqual(n.children, dict(right=right))
def test_node_add_then_delete_one_child(self): n = Node() left = Node("left") n.add_child(left) n.remove_child("left") self.assertEqual(n.children, dict())