ViperEkura
1c04a0b9fa
- serve_runtime.py and train_runtime.py are host-side Docker helpers, so they join train-entrypoint.sh and lib/ under scripts/docker/ - scripts/tools/ now contains only in-container CLIs - update wrapper call sites, test import, and docker guide references
83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
"""Unit tests for the serving runtime configuration parser."""
|
|
|
|
import pytest
|
|
|
|
from scripts.docker.serve_runtime import load_runtime
|
|
|
|
|
|
def _write(tmp_path, body: str) -> str:
|
|
config_path = tmp_path / "serve.yaml"
|
|
config_path.write_text(body, encoding="utf-8")
|
|
return str(config_path)
|
|
|
|
|
|
def test_runtime_exports_defaults(tmp_path):
|
|
config_path = _write(
|
|
tmp_path,
|
|
"runtime:\n"
|
|
" port: 8000\n"
|
|
" paths:\n"
|
|
" param: ./params\n"
|
|
"server:\n"
|
|
" device: cuda\n",
|
|
)
|
|
runtime = load_runtime(config_path)
|
|
assert runtime["SERVE_PORT"] == "8000"
|
|
assert runtime["SERVE_CONTAINER_PORT"] == "8000"
|
|
assert runtime["SERVE_PARAM_DIR"] == str((tmp_path / "params").resolve())
|
|
assert runtime["SERVE_GPU_ENABLED"] == "true"
|
|
assert "CUDA_VISIBLE_DEVICES" not in runtime
|
|
assert runtime["SERVE_DEVICE"] == "cuda"
|
|
assert runtime["CUDA_TAG"] == "cu128"
|
|
assert runtime["SERVE_JOB_NAME"] == ""
|
|
|
|
|
|
def test_runtime_gpu_disabled_requires_cpu(tmp_path):
|
|
config_path = _write(
|
|
tmp_path,
|
|
"runtime:\n gpu:\n enabled: false\nserver:\n device: cuda\n",
|
|
)
|
|
with pytest.raises(ValueError, match="server.device must be 'cpu'"):
|
|
load_runtime(config_path)
|
|
|
|
|
|
def test_runtime_gpu_devices_single_and_ports(tmp_path):
|
|
config_path = _write(
|
|
tmp_path,
|
|
"runtime:\n"
|
|
" gpu:\n"
|
|
" devices: [1]\n"
|
|
" port: 8080\n"
|
|
"server:\n"
|
|
" port: 9000\n"
|
|
" device: cuda\n",
|
|
)
|
|
runtime = load_runtime(config_path)
|
|
assert runtime["SERVE_PORT"] == "8080"
|
|
assert runtime["SERVE_CONTAINER_PORT"] == "9000"
|
|
assert runtime["CUDA_VISIBLE_DEVICES"] == "1"
|
|
|
|
|
|
def test_runtime_gpu_devices_rejects_multi(tmp_path):
|
|
config_path = _write(
|
|
tmp_path,
|
|
"runtime:\n gpu:\n devices: [0, 1]\n",
|
|
)
|
|
with pytest.raises(ValueError, match="single-device"):
|
|
load_runtime(config_path)
|
|
|
|
|
|
def test_runtime_port_out_of_range(tmp_path):
|
|
config_path = _write(tmp_path, "runtime:\n port: 70000\n")
|
|
with pytest.raises(ValueError, match="between 1 and 65535"):
|
|
load_runtime(config_path)
|
|
|
|
|
|
def test_runtime_environment_export(tmp_path):
|
|
config_path = _write(
|
|
tmp_path,
|
|
"runtime:\n environment:\n TOKENIZERS_PARALLELISM: 'false'\n",
|
|
)
|
|
runtime = load_runtime(config_path)
|
|
assert runtime["environment"] == {"TOKENIZERS_PARALLELISM": "false"}
|