def test_extract_max(self): heap = MaxHeap() heap.set_heap([16,14,10,8,7,9,3,2,4,1]) max_elem = heap.extract_max() self.assertEqual(max_elem, 16) self.assertEqual(heap.get_heap(), [14,8,10,4,7,9,3,2,1])
def test_extract_max(self): heap = MaxHeap() heap.set_heap([16, 14, 10, 8, 7, 9, 3, 2, 4, 1]) max_elem = heap.extract_max() self.assertEqual(max_elem, 16) self.assertEqual(heap.get_heap(), [14, 8, 10, 4, 7, 9, 3, 2, 1])