| 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" | |