test_diarize_errors.py

tests/test_diarize_errors.py

1. """Hugging Face access errors become instructions, not tracebacks."""
3. from transcription_pipeline.diarize import DEFAULT_MODEL as MODEL
4. from transcription_pipeline.diarize import explain_load_error
7. class GatedRepoError(Exception):
8. pass
11. def test_gated_model_says_where_to_accept_the_terms():
12. e = explain_load_error(GatedRepoError("403 Client Error. Cannot access gated repo"), MODEL)
13. assert e is not None
14. assert "has not accepted the terms" in str(e)
15. assert f"https://huggingface.co/{MODEL}" in str(e)
18. def test_matches_the_message_too():
19. e = explain_load_error(RuntimeError("you are not in the authorized list"), MODEL)
20. assert e is not None and "accept" in str(e)
23. def test_bad_token():
24. e = explain_load_error(RuntimeError("401 Client Error: Invalid credentials"), MODEL)
25. assert e is not None and "rejected the token" in str(e)
28. def test_other_errors_pass_through():
29. assert explain_load_error(OSError("disk full"), MODEL) is None