def test_max_withNotEmptyBST_shouldReturnMaxKeyFromBST(self): bst = AVLTree() bst.put("A", 1) self.assertEqual("A", bst.max()) bst.put("B", 2) self.assertEqual("B", bst.max()) bst.put("C", 3) self.assertEqual("C", bst.max()) bst.put("D", 4) self.assertEqual("D", bst.max()) bst.delete_max() self.assertEqual("C", bst.max()) bst.delete_max() self.assertEqual("B", bst.max()) bst.delete_max() self.assertEqual("A", bst.max())
def test_max_withCompareFunctionAndNotEmptyBST_shouldReturnMaxKeyFromBST( self): bst = AVLTree(self.cmp) bst.put([0, 1], 1) self.assertEqual([0, 1], bst.max()) bst.put([0, 2], 2) self.assertEqual([0, 2], bst.max()) bst.put([0, 3], 3) self.assertEqual([0, 3], bst.max()) bst.put([0, 4], 4) self.assertEqual([0, 4], bst.max()) bst.delete_max() self.assertEqual([0, 3], bst.max()) bst.delete_max() self.assertEqual([0, 2], bst.max()) bst.delete_max() self.assertEqual([0, 1], bst.max())