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())