diff options
| author | Damien George <damien@micropython.org> | 2021-05-08 18:36:27 +1000 |
|---|---|---|
| committer | Damien George <damien@micropython.org> | 2021-05-08 22:47:03 +1000 |
| commit | 7b923d6c72be29e2fe57740c9ea3a2135e7db6f7 (patch) | |
| tree | dac5cfa77b1b592ecabfffb4d019419e40de1a33 | |
| parent | 9340cfe77432b548fc6ace6e2959a07a08e8eadc (diff) | |
tests/thread: Make stress_aes.py test run on bare-metal ports.
This is a long-running test, so make it run in reasonable time on slower,
bare-metal ports.
Signed-off-by: Damien George <damien@micropython.org>
| -rw-r--r-- | tests/thread/stress_aes.py | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/tests/thread/stress_aes.py b/tests/thread/stress_aes.py index f73da557c..1bf252a80 100644 --- a/tests/thread/stress_aes.py +++ b/tests/thread/stress_aes.py @@ -235,7 +235,7 @@ class LockedCounter: count = LockedCounter() -def thread_entry(): +def thread_entry(n_loop): global count aes = AES(256) @@ -244,7 +244,7 @@ def thread_entry(): data = bytearray(128) # from now on we don't use the heap - for loop in range(5): + for loop in range(n_loop): # encrypt aes.set_key(key) aes.set_iv(iv) @@ -265,8 +265,20 @@ def thread_entry(): if __name__ == "__main__": - n_thread = 20 + import sys + + if sys.platform == "rp2": + n_thread = 1 + n_loop = 2 + elif sys.platform in ("esp32", "pyboard"): + n_thread = 2 + n_loop = 2 + else: + n_thread = 20 + n_loop = 5 for i in range(n_thread): - _thread.start_new_thread(thread_entry, ()) + _thread.start_new_thread(thread_entry, (n_loop,)) + thread_entry(n_loop) while count.value < n_thread: time.sleep(1) + print("done") |