mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Casper da Costa-Luis
276a2b02a3
* ui: default no_progress_bar=None instead of False Fixes #3314 * ui: catch remaining missing auto-quiet progress * progress: treat disable=False as None * test: progress.Tqdm(disable=False) * revert no_progress_bar=None <= False partially reverts e0342a8641f0101d
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
import logging
|
|
import sys
|
|
|
|
import mock
|
|
|
|
from dvc.progress import Tqdm
|
|
from dvc.utils import env2bool
|
|
|
|
|
|
def test_quiet_logging(caplog, capsys):
|
|
with caplog.at_level(logging.CRITICAL, logger="dvc"):
|
|
for _ in Tqdm(range(10)):
|
|
pass
|
|
out_err = capsys.readouterr()
|
|
assert out_err.out == ""
|
|
assert out_err.err == ""
|
|
|
|
|
|
def test_quiet_logging_disable_false(caplog, capsys):
|
|
with caplog.at_level(logging.CRITICAL, logger="dvc"):
|
|
# simulate interactive terminal
|
|
with mock.patch.object(sys.stderr, "isatty", return_value=True):
|
|
for _ in Tqdm(range(10), disable=False):
|
|
pass
|
|
out_err = capsys.readouterr()
|
|
assert out_err.out == ""
|
|
assert out_err.err == ""
|
|
|
|
|
|
def test_quiet_notty(caplog, capsys):
|
|
with caplog.at_level(logging.INFO, logger="dvc"):
|
|
for _ in Tqdm(range(10)):
|
|
pass
|
|
out_err = capsys.readouterr()
|
|
assert out_err.out == ""
|
|
if env2bool("DVC_IGNORE_ISATTY"):
|
|
assert "0/10" in out_err.err
|
|
else:
|
|
assert out_err.err == ""
|
|
|
|
|
|
def test_default(caplog, capsys):
|
|
with caplog.at_level(logging.INFO, logger="dvc"):
|
|
# simulate interactive terminal
|
|
with mock.patch.object(sys.stderr, "isatty", return_value=True):
|
|
for _ in Tqdm(range(10)):
|
|
pass
|
|
out_err = capsys.readouterr()
|
|
assert out_err.out == ""
|
|
assert "0/10" in out_err.err
|