Files
dvc/tests/unit/test_progress.py
T

15 lines
374 B
Python

import mock
from unittest import TestCase
from dvc.progress import ProgressCallback
class TestProgressCallback(TestCase):
@mock.patch("dvc.progress.progress")
def test_should_init_reset_progress(self, progress_mock):
total_files_num = 1
ProgressCallback(total_files_num)
self.assertEqual([mock.call.reset()], progress_mock.method_calls)