File size: 1,157 Bytes
3e66d5d a358eab 6c33365 a358eab 3e66d5d a358eab 3e66d5d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | """Tests for environment diagnostics."""
from facecloak.environment import collect_runtime_report, render_runtime_markdown
def test_collect_runtime_report_returns_expected_values() -> None:
report = collect_runtime_report()
assert report.status == "ready"
assert report.tensor_sanity == (1, 2, 3)
assert report.device == ("cuda" if report.cuda_available else "cpu")
assert report.torch_version.startswith("2.2.2")
assert report.facenet_pytorch_version == "2.6.0"
assert report.gradio_version == "6.12.0"
assert report.huggingface_hub_version == "1.10.2"
def test_collect_runtime_report_has_no_phase_field() -> None:
"""RuntimeReport must not contain a phase field after the refactor."""
report = collect_runtime_report()
assert not hasattr(report, "phase")
def test_render_runtime_markdown_contains_environment_summary() -> None:
markdown = render_runtime_markdown()
assert "Runtime Diagnostics" in markdown
assert "CUDA Available" in markdown
assert "Torch Cache" in markdown
# Must NOT mention phase numbers
assert "Phase 2" not in markdown
assert "Phase 3" not in markdown
|