tests/test_hardware_fallback_policy.py

from pathlib import Path

from src.jobs import normalize_auto_space_hardware


def test_restricted_hardware_not_offered_as_automatic_fallback():
    html = Path("web/index.html").read_text(encoding="utf-8")
    fallback_select = html.split('id="fallbackHardware"', 1)[1].split("</select>", 1)[0]
    assert "a100-large" not in fallback_select
    assert "h200" not in fallback_select
    assert "a10g-large" in fallback_select
    assert "l40sx1" in fallback_select


def test_backend_sanitizes_restricted_automatic_hardware():
    assert normalize_auto_space_hardware("a100-large", default="a10g-large") == "a10g-large"
    assert normalize_auto_space_hardware("h200", default="a10g-large") == "a10g-large"
    assert normalize_auto_space_hardware("l40sx1", default="a10g-large") == "l40sx1"