There was an error while loading. Please reload this page.
1 parent f9ae3e9 commit 343c970Copy full SHA for 343c970
1 file changed
scripts/torch_ci/test_entrypoint.py
@@ -1,9 +1,10 @@
1
"""Unit tests for entrypoint.py."""
2
3
-import unittest
4
-from unittest.mock import Mock
+import io
5
import pathlib
6
import shutil
+import unittest
7
+from unittest.mock import Mock, patch
8
9
from scripts.torch_ci.entrypoint import CoverageRunner, CommandExecutor
10
@@ -61,6 +62,12 @@ def test_format_results_table(self):
61
62
self.assertIn("op1", table)
63
self.assertIn("PASS", table)
64
65
+ @patch("sys.stdout", new_callable=io.StringIO)
66
+ def test_run_all_no_spam_stdout(self, mock_stdout):
67
+ self.mock_executor.run.return_value = Mock(returncode=0, stderr="")
68
+ self.runner.run_all()
69
+ self.assertNotIn("Running coverage for", mock_stdout.getvalue())
70
+
71
72
if __name__ == "__main__":
73
unittest.main()
0 commit comments