micropython/tests/extmod/uheapq1.py

41 wiersze
643 B
Python
Czysty Zwykły widok Historia

2014-10-22 17:37:18 +00:00
try:
import uheapq as heapq
except:
2017-02-14 22:56:22 +00:00
try:
import heapq
except ImportError:
print("SKIP")
raise SystemExit
2014-10-22 17:37:18 +00:00
try:
heapq.heappop([])
except IndexError:
print("IndexError")
try:
heapq.heappush((), 1)
except TypeError:
print("TypeError")
def pop_and_print(h):
l = []
while h:
l.append(str(heapq.heappop(h)))
print(' '.join(l))
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)