Can’t you use the -b or --buffer command-line option <https://docs.python.org/dev/library/unittest.html#cmdoption-unittest-b>, or equivalent option to unittest.main() or whatever?

The standard output and standard error streams are buffered during the test run. Output during a passing test is discarded. Output is echoed normally on test fail or error and is added to the failure messages.