ykcheungab
101 lines
3.4 KiB
Python
101 lines
3.4 KiB
Python
"""Post-build smoke tests for packaged wheels and standalone GUI builds."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import venv
|
|
from pathlib import Path
|
|
|
|
from paths import GUI_DIST_DIR
|
|
|
|
# Public SDK surface that must import from a released wheel.
|
|
_SMOKE_IMPORT_SCRIPT = """
|
|
import orisys
|
|
from orisys import Sensor, list_config, export_config, util, finger
|
|
from orisys.finger import UVMeshViewer, build_uv_surface_mesh, FingerRuntime
|
|
from orisys.finger.mesh import UVSurfaceMesh
|
|
from orisys.finger.viewer import init_uv_mesh_viewer
|
|
from orisys.hhd import nHHD
|
|
from orisys.tracker import Tracker
|
|
|
|
assert orisys.__version__
|
|
assert Sensor is not None
|
|
assert callable(list_config)
|
|
assert callable(export_config)
|
|
assert util is not None
|
|
assert finger is not None
|
|
assert UVMeshViewer is not None
|
|
assert build_uv_surface_mesh is not None
|
|
assert FingerRuntime is not None
|
|
assert UVSurfaceMesh is not None
|
|
assert init_uv_mesh_viewer is not None
|
|
assert nHHD is not None
|
|
assert Tracker is not None
|
|
print(f"orisys {orisys.__version__} public API imports OK")
|
|
"""
|
|
|
|
|
|
def _venv_python(venv_dir: Path) -> Path:
|
|
if sys.platform == "win32":
|
|
return venv_dir / "Scripts" / "python.exe"
|
|
return venv_dir / "bin" / "python"
|
|
|
|
|
|
def verify_built_wheel(wheel_path: Path) -> None:
|
|
"""Install the wheel in a temporary venv and smoke-test public imports."""
|
|
wheel_path = wheel_path.resolve()
|
|
with tempfile.TemporaryDirectory(prefix="orisys-wheel-verify-") as tmp:
|
|
venv_dir = Path(tmp) / "venv"
|
|
venv.create(venv_dir, with_pip=True)
|
|
python = _venv_python(venv_dir)
|
|
|
|
subprocess.check_call(
|
|
[python, "-m", "pip", "install", "-q", "--upgrade", "pip"],
|
|
)
|
|
subprocess.check_call(
|
|
[python, "-m", "pip", "install", "-q", str(wheel_path)],
|
|
)
|
|
subprocess.check_call([python, "-c", _SMOKE_IMPORT_SCRIPT])
|
|
|
|
|
|
def verify_gui_dist(dist_dir: Path | None = None) -> None:
|
|
"""Smoke-test a standalone GUI folder produced by build_gui_win.py."""
|
|
dist_dir = (dist_dir or GUI_DIST_DIR).resolve()
|
|
exe = dist_dir / "OrisysDashboard.exe"
|
|
if not exe.is_file():
|
|
raise FileNotFoundError(f"GUI executable not found: {exe}")
|
|
|
|
required = [
|
|
dist_dir / "config" / "dd02-ov.json",
|
|
dist_dir / "assets" / "finger" / "visualization" / "world_to_cam.npy",
|
|
dist_dir / "assets" / "finger" / "30fps-0703" / "mesh.npz",
|
|
]
|
|
missing = [str(path.relative_to(dist_dir)) for path in required if not path.is_file()]
|
|
if missing:
|
|
raise FileNotFoundError(f"Missing bundled files in {dist_dir}: {', '.join(missing)}")
|
|
|
|
subprocess.check_call([str(exe), "--help"], cwd=dist_dir)
|
|
print(f"GUI dist smoke test OK: {exe}")
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Verify packaged Orisys artifacts.")
|
|
parser.add_argument("--gui-dist", action="store_true", help="Verify dist/OrisysDashboard/")
|
|
parser.add_argument("--dist-dir", type=Path, default=None, help="Override GUI dist folder")
|
|
parser.add_argument("wheel", nargs="?", type=Path, help="Wheel path for import smoke test")
|
|
args = parser.parse_args()
|
|
|
|
if args.gui_dist:
|
|
verify_gui_dist(args.dist_dir)
|
|
return
|
|
if args.wheel is None:
|
|
parser.error("Provide a wheel path or pass --gui-dist")
|
|
verify_built_wheel(args.wheel)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|