def test_remove_removesElemAndMaintainsHeapProperty(self): minHeap = MinHeap() arr = [3, 9, 12, 13, 1] _ = minHeap.convertArrayToHeap(arr) minHeap.remove(13) expectedHeap = [1, 3, 12, 9] self.assertListEqual(expectedHeap, minHeap.heap)
def test_convertArrayToHeap_MaintainsHeapOrder(self): minHeap = MinHeap() arr = [3, 9, 12, 7, 1] expectedHeap = [1, 3, 12, 9, 7] res = minHeap.convertArrayToHeap(arr) self.assertListEqual(expectedHeap, res)