mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
update dvc.progress test stub
This commit is contained in:
+12
-14
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user