test_quiet_warnings.py

tests/test_quiet_warnings.py

1. """pyannote's torchcodec warning is silenced, in the CLI and under pytest."""
3. import warnings
5. from transcription_pipeline.cli import QUIET
7. # The shape pyannote emits: the message opens with a line break.
8. MESSAGE = "\ntorchcodec is not installed correctly so built-in audio decoding will fail."
11. def test_cli_filter_catches_the_real_message():
12. with warnings.catch_warnings(record=True) as caught:
13. warnings.simplefilter("always")
14. for action, message in QUIET:
15. warnings.filterwarnings(action, message=message)
16. warnings.warn(MESSAGE, UserWarning)
17. warnings.warn("something else", UserWarning)
18. assert [str(w.message) for w in caught] == ["something else"]
21. def test_pytest_config_catches_it_too():
22. # No filters of our own here: only pyproject's filterwarnings is in force.
23. with warnings.catch_warnings(record=True) as caught:
24. warnings.warn(MESSAGE, UserWarning)
25. warnings.warn("something else", UserWarning)
26. assert [str(w.message) for w in caught] == ["something else"]