test with uvloop

Author: minrk

setup.cfg

@@ -20,7 +20,9 @@ py_modules = asyncio_atexit
 python_requires = >=3.6
 
 [options.extras_require]
-test = pytest
+test =
+    pytest
+    uvloop
 
 [tool:pytest]
 # asyncio_mode = auto

test_asyncio_atexit.py

@@ -1,6 +1,13 @@
 import asyncio
 import sys
 
+try:
+    import uvloop
+except ImportError:
+    uvloop = None
+
+import pytest
+
 import asyncio_atexit
 
 if sys.version_info >= (3, 7):
@@ -15,7 +22,24 @@ def asyncio_run(coro):
             loop.close()
 
 
-def test_asyncio_atexit():
+policies = ["default"]
+if uvloop is not None:
+    policies.append("uvloop")
+
+
+@pytest.fixture(params=policies)
+def policy(request):
+    before_policy = asyncio.get_event_loop_policy()
+    if request.param == "default":
+        policy = asyncio.DefaultEventLoopPolicy()
+    elif request.param == "uvloop":
+        policy = uvloop.EventLoopPolicy()
+    asyncio.set_event_loop_policy(policy)
+    yield
+    asyncio.set_event_loop_policy(before_policy)
+
+
+def test_asyncio_atexit(policy):
     sync_called = False
     async_called = False
 
@@ -37,7 +61,7 @@ async def test():
     assert async_called
 
 
-def test_unregister():
+def test_unregister(policy):
     sync_called = False
 
     def sync_cb():