From f5d69794a86f7647de3f3b3efd282ed288b1cba6 Mon Sep 17 00:00:00 2001 From: Damien George Date: Wed, 22 Oct 2014 17:37:18 +0000 Subject: extmod: Add uheapq module. --- tests/extmod/uheapq1.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 tests/extmod/uheapq1.py (limited to 'tests/extmod') diff --git a/tests/extmod/uheapq1.py b/tests/extmod/uheapq1.py new file mode 100644 index 000000000..e71f817ef --- /dev/null +++ b/tests/extmod/uheapq1.py @@ -0,0 +1,36 @@ +try: + import uheapq as heapq +except: + import heapq + +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) -- cgit v1.2.3