update dvc.progress test stub

This commit is contained in:
Casper da Costa-Luis
2019-08-10 17:18:12 +01:00
parent 56db2ee5fe
commit 2ac99d87e5
+12 -14
View File
@@ -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