zhifu gao
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
EXAMPLE = ROOT / "examples" / "openai_api"
|
|
PLAN = ROOT / "docs" / "community_growth_20k.md"
|
|
|
|
|
|
def test_openai_api_docker_validation_script_documents_cpu_and_gpu_paths():
|
|
script = EXAMPLE / "validate_docker.sh"
|
|
text = script.read_text()
|
|
|
|
required = [
|
|
"docker build -t",
|
|
"docker run --rm -d",
|
|
"FUNASR_DEVICE=cpu",
|
|
"python smoke_test.py --base-url",
|
|
"--gpu",
|
|
"--gpus all",
|
|
"FUNASR_DEVICE=cuda",
|
|
"nvidia-smi",
|
|
"NVIDIA Container Toolkit",
|
|
]
|
|
for marker in required:
|
|
assert marker in text
|
|
|
|
|
|
def test_openai_api_readmes_link_docker_validation_script():
|
|
readmes = [
|
|
EXAMPLE / "README.md",
|
|
EXAMPLE / "README_zh.md",
|
|
]
|
|
|
|
for readme in readmes:
|
|
text = readme.read_text()
|
|
assert "validate_docker.sh" in text
|
|
assert "FUNASR_DEVICE=cpu" in text
|
|
assert "FUNASR_DEVICE=cuda" in text
|
|
|
|
|
|
def test_growth_plan_records_docker_validation_blocker_and_command():
|
|
text = PLAN.read_text()
|
|
|
|
required = [
|
|
"[ ] Validate Docker CPU and GPU images",
|
|
"`examples/openai_api/validate_docker.sh`",
|
|
"`bash validate_docker.sh`",
|
|
"`bash validate_docker.sh --gpu`",
|
|
"`ind-gpu8` has H100 GPUs but no `docker` CLI",
|
|
"`funasr-web` also has no `docker` CLI",
|
|
]
|
|
for marker in required:
|
|
assert marker in text
|