migrate to pytest

tests/unit/test_progress.py

@@ -0,0 +1,20 @@
+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")