chore: import upstream snapshot with attribution

tests/python/sdk/test_progress.py

@@ -0,0 +1,81 @@
+# Copyright (C) CVAT.ai Corporation
+#
+# SPDX-License-Identifier: MIT
+
+import io
+import warnings
+
+import tqdm
+from cvat_sdk.core.helpers import DeferredTqdmProgressReporter, TqdmProgressReporter
+from cvat_sdk.core.progress import NullProgressReporter, ProgressReporter
+
+
+def _exercise_reporter(r: ProgressReporter) -> None:
+    with r.task(total=5, desc="Test task", unit="parrots"):
+        r.advance(1)
+        r.report_status(4)
+
+        for x in r.iter(["x"]):
+            assert x == "x"
+
+
+def test_null_reporter():
+    _exercise_reporter(NullProgressReporter())
+    # NPR doesn't do anything, so there's nothing to assert
+
+
+def test_tqdm_reporter():
+    f = io.StringIO()
+
+    instance = tqdm.tqdm(file=f)
+
+    with warnings.catch_warnings():
+        r = TqdmProgressReporter(instance)
+
+    _exercise_reporter(r)
+
+    output = f.getvalue()
+
+    assert "100%" in output
+    assert "Test task" in output
+    # TPR doesn't support parameters other than "total" and "desc",
+    # so there won't be any parrots in the output.
+
+
+def test_deferred_tqdm_reporter():
+    f = io.StringIO()
+
+    _exercise_reporter(DeferredTqdmProgressReporter({"file": f}))
+
+    output = f.getvalue()
+
+    assert "100%" in output
+    assert "Test task" in output
+    assert "parrots" in output
+
+
+class _LegacyProgressReporter(ProgressReporter):
+    # overriding start instead of start2
+    def start(self, total: int, *, desc: str | None = None) -> None:
+        self.total = total
+        self.desc = desc
+        self.progress = 0
+
+    def report_status(self, progress: int):
+        self.progress = progress
+
+    def advance(self, delta: int):
+        self.progress += delta
+
+    def finish(self):
+        self.finished = True
+
+
+def test_legacy_progress_reporter():
+    r = _LegacyProgressReporter()
+
+    _exercise_reporter(r)
+
+    assert r.total == 5
+    assert r.desc == "Test task"
+    assert r.progress == 5