def test_is_balanced(self):
tree = AVLTree(4, None, None)
self.assertTrue(tree.is_balanced())
tree = AVLTree(None, AVLTree(None, AVLTree(None, None, None), None),
None)
self.assertFalse(tree.is_balanced())
tree = AVLTree(
50, AVLTree(17, AVLTree(12, None, None), AVLTree(23, None, None)),
AVLTree(72, AVLTree(54, None, None), AVLTree(76, None, None)))
self.assertTrue(tree.is_balanced())
def test_remove_add_balanced(self):
tree = AVLTree(4, None, None)
tree.add(1).add(2).add(3).add(4).add(7).add(9).pop(4).pop(9).add(
20).add(-2).add(21).add(-3).pop(-3)
self.assertTrue(tree.is_balanced())