ZhanshenzhuiandSy03
48ae412a2f
Signed-off-by: Zhanshenzhui <99891002+zyforsure@users.noreply.github.com> Co-authored-by: Sy03 <1370724210@qq.com>
99 lines
2.8 KiB
Python
99 lines
2.8 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
pytestmark = [pytest.mark.core_model, pytest.mark.cpu, pytest.mark.example]
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[3]
|
|
EXAMPLE_SCRIPT = REPO_ROOT / "examples" / "offline_inference" / "mage_vl" / "end2end.py"
|
|
|
|
|
|
def _fake_mage_repo(tmp_path: Path) -> Path:
|
|
repo = tmp_path / "Mage"
|
|
mage_vl = repo / "mage_vl"
|
|
mage_vl.mkdir(parents=True)
|
|
(mage_vl / "inference_base.py").write_text("print('base')\n", encoding="utf-8")
|
|
(mage_vl / "inference_streaming.py").write_text("print('streaming')\n", encoding="utf-8")
|
|
return repo
|
|
|
|
|
|
def test_mage_vl_offline_example_builds_codec_command(tmp_path: Path) -> None:
|
|
mage_repo = _fake_mage_repo(tmp_path)
|
|
video_path = tmp_path / "input.mp4"
|
|
video_path.write_bytes(b"fake")
|
|
output_dir = tmp_path / "outputs"
|
|
|
|
subprocess.run(
|
|
[
|
|
sys.executable,
|
|
str(EXAMPLE_SCRIPT),
|
|
"--mage-repo",
|
|
str(mage_repo),
|
|
"--task",
|
|
"video",
|
|
"--video-path",
|
|
str(video_path),
|
|
"--video-backend",
|
|
"codec",
|
|
"--codec-engine",
|
|
"traditional",
|
|
"--output-dir",
|
|
str(output_dir),
|
|
"--dry-run",
|
|
],
|
|
check=True,
|
|
)
|
|
|
|
with open(output_dir / "summary.json", encoding="utf-8") as f:
|
|
summary = json.load(f)
|
|
|
|
assert summary["task"] == "video"
|
|
assert summary["dry_run"] is True
|
|
assert "--video-backend" in summary["command"]
|
|
assert "codec" in summary["command"]
|
|
assert "--codec-engine" in summary["command"]
|
|
assert "traditional" in summary["command"]
|
|
|
|
|
|
def test_mage_vl_offline_example_builds_streaming_command(tmp_path: Path) -> None:
|
|
mage_repo = _fake_mage_repo(tmp_path)
|
|
video_path = tmp_path / "input.mp4"
|
|
video_path.write_bytes(b"fake")
|
|
output_dir = tmp_path / "outputs"
|
|
|
|
subprocess.run(
|
|
[
|
|
sys.executable,
|
|
str(EXAMPLE_SCRIPT),
|
|
"--mage-repo",
|
|
str(mage_repo),
|
|
"--task",
|
|
"streaming",
|
|
"--video-path",
|
|
str(video_path),
|
|
"--max-segments",
|
|
"1",
|
|
"--output-dir",
|
|
str(output_dir),
|
|
"--dry-run",
|
|
],
|
|
check=True,
|
|
)
|
|
|
|
with open(output_dir / "summary.json", encoding="utf-8") as f:
|
|
summary = json.load(f)
|
|
|
|
command = summary["command"]
|
|
assert summary["task"] == "streaming"
|
|
assert str(mage_repo / "mage_vl" / "inference_streaming.py") in command
|
|
assert "--checkpoint" in command
|
|
assert "--max_segments" in command
|