def test_pop_heap():
heap1 = Heap(num_list)
heap2 = Heap()
heap3 = Heap()
heap2.extend(num_list)
for i in num_list:
heap3.push(i)
expected = [-2, 3, 10]
assert (expected == [i for i in pop_heap(heap1)])
assert (expected == [i for i in pop_heap(heap2)])
assert (expected == [i for i in pop_heap(heap3)])
def test_pop_heap():
heap1 = Heap(num_list)
heap2 = Heap()
heap3 = Heap()
heap2.extend(num_list)
for i in num_list:
heap3.push(i)
expected = [-2, 3, 10]
assert(expected == [i for i in pop_heap(heap1)])
assert(expected == [i for i in pop_heap(heap2)])
assert(expected == [i for i in pop_heap(heap3)])
