circuitpython/tests/extmod/uheapq1.py

43 lines
645 B
Python
Raw Normal View History

2014-10-22 13:37:18 -04:00
try:
import uheapq as heapq
except:
2017-02-14 17:56:22 -05:00
try:
import heapq
except ImportError:
print("SKIP")
raise SystemExit
2014-10-22 13:37:18 -04:00
try:
heapq.heappop([])
except IndexError:
print("IndexError")
try:
heapq.heappush((), 1)
except TypeError:
print("TypeError")
2014-10-22 13:37:18 -04:00
def pop_and_print(h):
l = []
while h:
l.append(str(heapq.heappop(h)))
print(" ".join(l))
2014-10-22 13:37:18 -04:00
h = []
heapq.heappush(h, 3)
heapq.heappush(h, 1)
heapq.heappush(h, 2)
print(h)
pop_and_print(h)
h = [4, 3, 8, 9, 10, 2, 7, 11, 5]
heapq.heapify(h)
print(h)
heapq.heappush(h, 1)
heapq.heappush(h, 6)
heapq.heappush(h, 12)
print(h)
pop_and_print(h)