3dc324d3f1
This adds the Python files in the tests/ directory to be formatted with ./tools/codeformat.py. The basics/ subdirectory is excluded for now so we aren't changing too much at once. In a few places `# fmt: off`/`# fmt: on` was used where the code had special formatting for readability or where the test was actually testing the specific formatting.
43 lines
645 B
Python
43 lines
645 B
Python
try:
|
|
import uheapq as heapq
|
|
except:
|
|
try:
|
|
import heapq
|
|
except ImportError:
|
|
print("SKIP")
|
|
raise SystemExit
|
|
|
|
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)
|