diff --git a/tests/unit/test_progress.py b/tests/unit/test_progress.py index 802bcb5fb..6e1a0cf81 100644 --- a/tests/unit/test_progress.py +++ b/tests/unit/test_progress.py @@ -1,19 +1,17 @@ import logging -import mock -from dvc.progress import progress, ProgressCallback +from dvc.progress import Tqdm def test_quiet(caplog, capsys): with caplog.at_level(logging.CRITICAL, logger="dvc"): - progress.clearln() - assert capsys.readouterr().out == "" - - -class TestProgressCallback: - @mock.patch("dvc.progress.progress") - def test_should_init_reset_progress(self, progress_mock): - total_files_num = 1 - - ProgressCallback(total_files_num) - - assert [mock.call.reset()] == progress_mock.method_calls + for _ in Tqdm(range(10)): + pass + out_err = capsys.readouterr() + assert out_err.out == "" + assert out_err.err == "" + with caplog.at_level(logging.INFO, logger="dvc"): + for _ in Tqdm(range(10)): + pass + out_err = capsys.readouterr() + assert out_err.out == "" + assert '0/10' in out_err.err