Пример #1
0
def test_min_heap():
    a = np.random.choice(1000, 200, replace=True).tolist()
    b = sorted(a)
    """ test insert """
    heap = MinHeap()
    for x in a:
        heap.insert(x)
    c = []
    while heap:
        c.append(heap.pop().key)
    assert b == c
    """ test heapify """
    heap = MinHeap(a)
    c = []
    while heap:
        c.append(heap.pop().key)
    assert b == c