mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
21 lines
534 B
Python
21 lines
534 B
Python
import mock
|
|
from unittest import TestCase
|
|
|
|
from dvc.progress import progress, progress_aware
|
|
|
|
|
|
class TestProgressAware(TestCase):
|
|
@mock.patch("sys.stdout.isatty", return_value=True)
|
|
@mock.patch("dvc.progress.Progress._print")
|
|
def test(self, mock_print, mock_isatty):
|
|
function = progress_aware(lambda: None)
|
|
|
|
function()
|
|
mock_print.assert_not_called()
|
|
|
|
progress.update_target("testing", 0, 100)
|
|
function()
|
|
mock_print.assert_called()
|
|
|
|
progress.finish_target("testing")
|