![Damien George](/assets/img/avatar_default.png)
The aim of this commit is to make it so that the existing thread tests can be used to test the _thread module on the rp2 port. The rp2 port only allows up to one thread to be created at a time, and does not have the GIL enabled. The following changes have been made: - run-tests.py skips mutation tests on rp2, because there's no GIL. - run-tests.py skips other tests on rp2 that require more than one thread. - The tests stop trying to start a new thread after there is an OSError, which indicates that the system cannot create more threads. - Some of these tests also now run the test function on the main thread, not just the spawned threads. - In some tests the output printing is adjusted so it's the same regardless of how many threads were spawned. - Some time.sleep(1) are replaced with time.sleep(0) to make the tests run a little faster (finish sooner when the work is done). For the most part the tests are unchanged for existing platforms like esp32 and unix. Signed-off-by: Damien George <damien@micropython.org>
45 lines
869 B
Python
45 lines
869 B
Python
# test threads sleeping
|
|
#
|
|
# MIT license; Copyright (c) 2016 Damien P. George on behalf of Pycom Ltd
|
|
|
|
import time
|
|
|
|
if hasattr(time, "sleep_ms"):
|
|
sleep_ms = time.sleep_ms
|
|
else:
|
|
sleep_ms = lambda t: time.sleep(t / 1000)
|
|
|
|
import _thread
|
|
|
|
lock = _thread.allocate_lock()
|
|
n_thread = 0
|
|
n_thread_max = 4
|
|
n_finished = 0
|
|
|
|
|
|
def thread_entry(t):
|
|
global n_finished
|
|
sleep_ms(t)
|
|
sleep_ms(2 * t)
|
|
with lock:
|
|
n_finished += 1
|
|
|
|
|
|
# spawn threads
|
|
for _ in range(n_thread_max):
|
|
try:
|
|
_thread.start_new_thread(thread_entry, (10 * n_thread,))
|
|
n_thread += 1
|
|
except OSError:
|
|
# System cannot create a new thead, so stop trying to create them.
|
|
break
|
|
|
|
# also run the function on this main thread
|
|
thread_entry(10 * n_thread)
|
|
n_thread += 1
|
|
|
|
# wait for threads to finish
|
|
while n_finished < n_thread:
|
|
sleep_ms(100)
|
|
print("done")
|