b070765427
The native emitter will not release/bounce the GIL when running code, so if it runs tight loops then no other threads get a chance to run (if the GIL is enabled). So for the thread tests, explicitly include a call to `time.sleep(0)` (or equivalent) to bounce the GIL and give other threads a chance to run. For some tests (eg `thread_coop.py`) the whole point of the test is to test that the GIL is correctly bounced. So for those cases force the use of the bytecode emitter for the busy functions. Signed-off-by: Damien George <damien@micropython.org>
56 lines
1.1 KiB
Python
56 lines
1.1 KiB
Python
# test setting the thread stack size
|
|
#
|
|
# MIT license; Copyright (c) 2016 Damien P. George on behalf of Pycom Ltd
|
|
|
|
import sys
|
|
import time
|
|
import _thread
|
|
|
|
# different implementations have different minimum sizes
|
|
if sys.implementation.name == "micropython":
|
|
sz = 2 * 1024
|
|
else:
|
|
sz = 512 * 1024
|
|
|
|
|
|
def foo():
|
|
pass
|
|
|
|
|
|
def thread_entry():
|
|
foo()
|
|
with lock:
|
|
global n_finished
|
|
n_finished += 1
|
|
|
|
|
|
# reset stack size to default
|
|
_thread.stack_size()
|
|
|
|
# test set/get of stack size
|
|
print(_thread.stack_size())
|
|
print(_thread.stack_size(sz))
|
|
print(_thread.stack_size() == sz)
|
|
print(_thread.stack_size())
|
|
|
|
lock = _thread.allocate_lock()
|
|
n_thread = 2
|
|
n_finished = 0
|
|
|
|
# set stack size and spawn a few threads
|
|
_thread.stack_size(sz)
|
|
for i in range(n_thread):
|
|
while True:
|
|
try:
|
|
_thread.start_new_thread(thread_entry, ())
|
|
break
|
|
except OSError:
|
|
pass
|
|
|
|
# reset stack size to default (for subsequent scripts on baremetal)
|
|
_thread.stack_size()
|
|
|
|
# busy wait for threads to finish
|
|
while n_finished < n_thread:
|
|
time.sleep(0)
|
|
print("done")
|