diff --git a/tests/test_utils_run_later.py b/tests/test_utils_run_later.py index 23c1c6263..cd39730d0 100644 --- a/tests/test_utils_run_later.py +++ b/tests/test_utils_run_later.py @@ -1,5 +1,6 @@ import pytest import time_machine +import io from datetime import datetime, timedelta from slic.utils.run_later import run_at, run_in, run_later, today, tomorrow, yesterday @@ -122,13 +123,14 @@ def test_run_at_tqdm_multiple_updates(monkeypatch, capsys): run_at(when, fexample) captured = capsys.readouterr() - captured_text = output.getvalue() assert "0%| |" in captured.err assert "50%|█████ |" in captured.err assert "100%|██████████|" in captured.err - assert "Function done!" in captured_text + assert "Function done!" in captured.out + + # Test today/yesterday and tomorrow functions