diff --git a/tests/stream/output/__init__.py b/tests/stream/output/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/stream/test_example.py b/tests/stream/output/test_example.py similarity index 100% rename from tests/stream/test_example.py rename to tests/stream/output/test_example.py diff --git a/tests/stream/test_print.py b/tests/stream/output/test_print.py similarity index 98% rename from tests/stream/test_print.py rename to tests/stream/output/test_print.py index 6e38766..7cf5cdc 100644 --- a/tests/stream/test_print.py +++ b/tests/stream/output/test_print.py @@ -7,8 +7,7 @@ from pytest import MonkeyPatch from atpbar.stream import FD, OutputStream - -from .st import st_text +from tests.stream.st import st_text def st_end() -> st.SearchStrategy[str | None]: diff --git a/tests/stream/test_stream.py b/tests/stream/output/test_stateful.py similarity index 94% rename from tests/stream/test_stream.py rename to tests/stream/output/test_stateful.py index f859010..5e419e0 100644 --- a/tests/stream/test_stream.py +++ b/tests/stream/output/test_stateful.py @@ -4,8 +4,7 @@ from hypothesis import strategies as st from atpbar.stream import OutputStream, Queue, StreamQueue - -from .st import st_text +from tests.stream.st import st_text class StatefulTest: @@ -41,7 +40,7 @@ def flush(self) -> None: @settings(max_examples=200) @given(data=st.data()) -def test_stream(data: st.DataObject) -> None: +def test_stateful(data: st.DataObject) -> None: test = StatefulTest(data=data) METHODS = [test.write, test.write_with_newline, test.flush] methods = data.draw(st.lists(st.sampled_from(METHODS)))