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