refac: move hypothesis code into integration/

Author: willemt

pytest_asyncio_cooperative/integration/__init__.py

@@ -0,0 +1,42 @@
+import asyncio
+import functools
+import time
+
+from ..fixtures import fill_fixtures
+
+
+async def hypothesis_test_wrapper(item):
+    """
+    Hypothesis is synchronous, let's run inside an executor to keep asynchronicity
+    """
+
+    # Do setup
+    item.start_setup = time.time()
+    fixture_values, teardowns = await fill_fixtures(item)
+    item.stop_setup = time.time()
+
+    default_loop = asyncio.get_running_loop()
+    inner_test = item.function.hypothesis.inner_test
+
+    def async_to_sync(*args, **kwargs):
+        # FIXME: can we cache this loop across multiple runs?
+        loop = asyncio.new_event_loop()
+        task = inner_test(*args, **kwargs)
+        try:
+            loop.run_until_complete(task)
+        finally:
+            loop.close()
+
+    # Run test
+    item.function.hypothesis.inner_test = async_to_sync
+    wrapped_func_with_fixtures = functools.partial(item.function, *fixture_values)
+    await default_loop.run_in_executor(None, wrapped_func_with_fixtures)
+
+    # Do teardowns
+    item.start_teardown = time.time()
+    for teardown in teardowns:
+        try:
+            await teardown.__anext__()
+        except StopAsyncIteration:
+            pass
+    item.stop_teardown = time.time()

pytest_asyncio_cooperative/integration/hypothesis.py

@@ -1,6 +1,5 @@
 import asyncio
 import collections.abc
-import functools
 import inspect
 import threading
 import time
@@ -12,6 +11,7 @@
 
 from .assertion import activate_assert_rewrite
 from .fixtures import fill_fixtures
+from .integration.hypothesis import hypothesis_test_wrapper
 
 
 def pytest_addoption(parser):
@@ -115,44 +115,6 @@ async def do_teardowns():
     await do_teardowns()
 
 
-# TODO: move to hypothesis module
-async def hypothesis_test_wrapper(item):
-    """
-    Hypothesis is synchronous, let's run inside an executor to keep asynchronicity
-    """
-
-    # Do setup
-    item.start_setup = time.time()
-    fixture_values, teardowns = await fill_fixtures(item)
-    item.stop_setup = time.time()
-
-    default_loop = asyncio.get_running_loop()
-    inner_test = item.function.hypothesis.inner_test
-
-    def async_to_sync(*args, **kwargs):
-        # FIXME: can we cache this loop across multiple runs?
-        loop = asyncio.new_event_loop()
-        task = inner_test(*args, **kwargs)
-        try:
-            loop.run_until_complete(task)
-        finally:
-            loop.close()
-
-    # Run test
-    item.function.hypothesis.inner_test = async_to_sync
-    wrapped_func_with_fixtures = functools.partial(item.function, *fixture_values)
-    await default_loop.run_in_executor(None, wrapped_func_with_fixtures)
-
-    # Do teardowns
-    item.start_teardown = time.time()
-    for teardown in teardowns:
-        try:
-            await teardown.__anext__()
-        except StopAsyncIteration:
-            pass
-    item.stop_teardown = time.time()
-
-
 def item_to_task(item):
     if getattr(item.function, "is_hypothesis_test", False):
         return hypothesis_test_wrapper(item)