From 512a2f25a241a901217ff3307e8228354b41be55 Mon Sep 17 00:00:00 2001 From: Tyler Goodlet Date: Mon, 26 Nov 2018 11:26:04 -0500 Subject: [PATCH] Expose `tractor_test` in the same way as `trio` --- tests/conftest.py | 2 +- tractor/testing/__init__.py | 1 + tractor/{ => testing}/_tractor_test.py | 5 ++++- 3 files changed, 6 insertions(+), 2 deletions(-) create mode 100644 tractor/testing/__init__.py rename tractor/{ => testing}/_tractor_test.py (92%) diff --git a/tests/conftest.py b/tests/conftest.py index a84205f..97dccd7 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -5,7 +5,7 @@ import random import pytest import tractor -from tractor._tractor_test import tractor_test +from tractor.testing import tractor_test pytest_plugins = ['pytester'] diff --git a/tractor/testing/__init__.py b/tractor/testing/__init__.py new file mode 100644 index 0000000..8bcfa90 --- /dev/null +++ b/tractor/testing/__init__.py @@ -0,0 +1 @@ +from ._tractor_test import tractor_test diff --git a/tractor/_tractor_test.py b/tractor/testing/_tractor_test.py similarity index 92% rename from tractor/_tractor_test.py rename to tractor/testing/_tractor_test.py index 077570a..05e0b24 100644 --- a/tractor/_tractor_test.py +++ b/tractor/testing/_tractor_test.py @@ -1,7 +1,10 @@ import inspect from functools import partial, wraps -from . import run +from .. import run + + +__all__ = ['tractor_test'] def tractor_test(fn):