2
2
import trio_asyncio
3
3
import asyncio
4
4
import pytest
5
+ from trio_asyncio ._loop import TrioPolicy
5
6
6
7
# Tests for concurrent or nested loops
7
8
@@ -46,7 +47,7 @@ async def gen_loop(i, task_status=trio.TASK_STATUS_IGNORED):
46
47
47
48
48
49
async def _test_same_task ():
49
- assert isinstance (asyncio .get_event_loop_policy (), trio_asyncio . TrioPolicy )
50
+ assert isinstance (asyncio .get_event_loop_policy (), TrioPolicy )
50
51
51
52
def get_loop (i , loop , policy ):
52
53
assert loop == asyncio .get_event_loop ()
@@ -55,18 +56,18 @@ def get_loop(i, loop, policy):
55
56
async with trio .open_nursery ():
56
57
async with trio_asyncio .open_loop () as loop1 :
57
58
policy = asyncio .get_event_loop_policy ()
58
- assert isinstance (policy , trio_asyncio . TrioPolicy )
59
+ assert isinstance (policy , TrioPolicy )
59
60
async with trio_asyncio .open_loop () as loop2 :
60
61
p2 = asyncio .get_event_loop_policy ()
61
62
assert policy is p2 , (policy , p2 )
62
63
loop1 .call_later (0.1 , get_loop , 0 , loop1 , policy )
63
64
loop2 .call_later (0.1 , get_loop , 1 , loop2 , policy )
64
65
await trio .sleep (0.2 )
65
66
66
- assert isinstance (asyncio .get_event_loop_policy (), trio_asyncio . TrioPolicy )
67
+ assert isinstance (asyncio .get_event_loop_policy (), TrioPolicy )
67
68
assert asyncio ._get_running_loop () is None
68
69
69
70
70
71
def test_same_task ():
71
- assert not isinstance (asyncio .get_event_loop_policy (), trio_asyncio . TrioPolicy )
72
+ assert not isinstance (asyncio .get_event_loop_policy (), TrioPolicy )
72
73
trio .run (_test_same_task )
0 commit comments