Example #1
0
    def test_insert_and_del_max_from_algs_book(self):
        heap = MaxHeap(['E', 'P', 'I', 'S', 'N', 'H', 'G', 'R', 'O', 'A'])
        heap.insert('T')

        self.assertEqual(heap.del_max(), 'T')
        self.assertEqual(heap.del_max(), 'S')
        self.assertEqual(heap.del_max(), 'R')
        self.assertEqual(heap.del_max(), 'P')
Example #2
0
    def test_insert_and_del_max(self):
        heap = MaxHeap([1, 2])
        heap.insert(4)
        heap.insert(3)

        self.assertEqual(heap.size, 4)

        self.assertEqual(heap.del_max(), 4)
        self.assertEqual(heap.del_max(), 3)
        self.assertEqual(heap.del_max(), 2)
        self.assertEqual(heap.del_max(), 1)