3dc324d3f1
This adds the Python files in the tests/ directory to be formatted with ./tools/codeformat.py. The basics/ subdirectory is excluded for now so we aren't changing too much at once. In a few places `# fmt: off`/`# fmt: on` was used where the code had special formatting for readability or where the test was actually testing the specific formatting.
36 lines
715 B
Python
36 lines
715 B
Python
# test capability for threads to access a shared mutable data structure
|
|
# (without contention because they access different parts of the structure)
|
|
#
|
|
# MIT license; Copyright (c) 2016 Damien P. George on behalf of Pycom Ltd
|
|
|
|
import _thread
|
|
|
|
|
|
def foo(lst, i):
|
|
lst[i] += 1
|
|
|
|
|
|
def thread_entry(n, lst, idx):
|
|
for i in range(n):
|
|
foo(lst, idx)
|
|
with lock:
|
|
global n_finished
|
|
n_finished += 1
|
|
|
|
|
|
lock = _thread.allocate_lock()
|
|
n_thread = 2
|
|
n_finished = 0
|
|
|
|
# the shared data structure
|
|
lst = [0, 0]
|
|
|
|
# spawn threads
|
|
for i in range(n_thread):
|
|
_thread.start_new_thread(thread_entry, ((i + 1) * 10, lst, i))
|
|
|
|
# busy wait for threads to finish
|
|
while n_finished < n_thread:
|
|
pass
|
|
print(lst)
|