diff --git a/tests/func/test_remote.py b/tests/func/test_remote.py index dadae44ff..a85aa4ae5 100644 --- a/tests/func/test_remote.py +++ b/tests/func/test_remote.py @@ -162,15 +162,16 @@ class TestRemoteShouldHandleUppercaseRemoteName(TestDvc): def test_large_dir_progress(repo_dir, dvc_repo): from dvc.utils import LARGE_DIR_SIZE - from dvc.progress import progress + from dvc.progress import Tqdm # Create a "large dir" for i in range(LARGE_DIR_SIZE + 1): repo_dir.create(os.path.join("gen", "{}.txt".format(i)), str(i)) - with patch.object(progress, "update_target") as update_target: + with patch.object(Tqdm, "truncate") as truncate: + assert not truncate.called dvc_repo.add("gen") - assert update_target.called + assert truncate.called def test_dir_checksum_should_be_key_order_agnostic(dvc_repo): diff --git a/tests/unit/test_progress.py b/tests/unit/test_progress.py index 6e1a0cf81..8149300c0 100644 --- a/tests/unit/test_progress.py +++ b/tests/unit/test_progress.py @@ -14,4 +14,4 @@ def test_quiet(caplog, capsys): pass out_err = capsys.readouterr() assert out_err.out == "" - assert '0/10' in out_err.err + assert "0/10" in out_err.err