@@ 31-47 (lines=17) @@ | ||
28 | assert run(parallele(children=[a_func, b_func, failure_func], succes_threshold=2)) |
|
29 | ||
30 | ||
31 | def test_run_curio(): |
|
32 | counter = ContextVar("counter", default=5) |
|
33 | ||
34 | async def reset_counter(): |
|
35 | if counter.get() == 5: |
|
36 | counter.set(0) |
|
37 | return 0 |
|
38 | return -1 |
|
39 | ||
40 | with Kernel() as kernel: |
|
41 | assert run_curio(kernel, reset_counter) == 0 |
|
42 | assert run_curio(kernel, reset_counter) == 0 |
|
43 | assert counter.get() == 5 |
|
44 | ||
45 | assert run(reset_counter) == 0 |
|
46 | assert run(reset_counter) == -1 |
|
47 | assert counter.get() == 0 |
|
48 |
@@ 8-24 (lines=17) @@ | ||
5 | from async_btree import run |
|
6 | ||
7 | ||
8 | def test_run_curio(): |
|
9 | counter = ContextVar("counter", default=5) |
|
10 | ||
11 | async def reset_counter(): |
|
12 | if counter.get() == 5: |
|
13 | counter.set(0) |
|
14 | return 0 |
|
15 | return -1 |
|
16 | ||
17 | with Kernel() as kernel: |
|
18 | assert run(kernel, reset_counter) == 0 |
|
19 | assert run(kernel, reset_counter) == 0 |
|
20 | assert counter.get() == 5 |
|
21 | ||
22 | assert _curio_run(reset_counter) == 0 |
|
23 | assert _curio_run(reset_counter) == -1 |
|
24 | assert counter.get() == 0 |
|
25 |