micropython/tests/extmod/uasyncio_heaplock.py
Damien George c4935f3049 tests/extmod: Add uasyncio tests.
All .exp files are included because they require CPython 3.8 which may not
always be available.
2020-03-26 01:25:45 +11:00

47 lines
919 B
Python

# test that basic scheduling of tasks, and uasyncio.sleep_ms, does not use the heap
import micropython
# strict stackless builds can't call functions without allocating a frame on the heap
try:
f = lambda: 0
micropython.heap_lock()
f()
micropython.heap_unlock()
except RuntimeError:
print("SKIP")
raise SystemExit
try:
import uasyncio as asyncio
except ImportError:
try:
import asyncio
except ImportError:
print("SKIP")
raise SystemExit
async def task(id, n, t):
for i in range(n):
print(id, i)
await asyncio.sleep_ms(t)
async def main():
t1 = asyncio.create_task(task(1, 4, 10))
t2 = asyncio.create_task(task(2, 4, 25))
micropython.heap_lock()
print("start")
await asyncio.sleep_ms(1)
print("sleep")
await asyncio.sleep_ms(100)
print("finish")
micropython.heap_unlock()
asyncio.run(main())