File size: 793 Bytes
4cf7d37 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | 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"
|