diff --git a/tests/test_pubsub.py b/tests/test_pubsub.py index 390ca29..ababcb5 100644 --- a/tests/test_pubsub.py +++ b/tests/test_pubsub.py @@ -4,9 +4,10 @@ from itertools import cycle import pytest import trio import tractor -from tractor.testing import tractor_test from tractor.experimental import msgpub +from conftest import tractor_test + def test_type_checks(): diff --git a/tests/test_streaming.py b/tests/test_streaming.py index 0d3ee1f..4e54e02 100644 --- a/tests/test_streaming.py +++ b/tests/test_streaming.py @@ -7,9 +7,10 @@ import platform import trio import tractor -from tractor.testing import tractor_test import pytest +from conftest import tractor_test + def test_must_define_ctx():