new test

tests/units/tokens/abstract/test_coroutine_wrapper.py

@@ -0,0 +1,28 @@
+import io
+import sys
+from contextlib import redirect_stdout
+
+import pytest
+
+from cantok import SimpleToken, TimeoutToken, ConditionToken, CounterToken
+
+
+@pytest.mark.parametrize(
+    ['create_value', 'expected_string'],
+    [
+        (lambda: SimpleToken(cancelled=True).wait(), ''),
+        (lambda: TimeoutToken(0.0001).wait(), ''),
+        (lambda: ConditionToken(lambda: True).wait(), ''),
+        (lambda: CounterToken(0).wait(), ''),
+        (lambda: 1, '1\n'),
+        (lambda: 'kek', f'{repr('kek')}\n'),
+    ],
+)
+def test_displayhook_printing_coroutine_wrappers_and_other_objects(create_value, expected_string):
+    buffer = io.StringIO()
+    with redirect_stdout(buffer):
+        sys.displayhook(create_value())
+
+    output = buffer.getvalue()
+
+    assert output == expected_string