#!/usr/bin/env python3
"""Verify that EDGS CUDA extension wheels are installed and importable."""

from __future__ import annotations

import importlib
import sys


def check_module(name: str) -> bool:
    try:
        module = importlib.import_module(name)
        print(f"OK: {name} -> {getattr(module, '__file__', '<builtin>')}")
        return True
    except Exception as error:
        print(f"FAIL: {name}: {error!r}")
        return False


def main() -> int:
    names = [
        "diff_gaussian_rasterization",
        "diff_gaussian_rasterization._C",
        "simple_knn._C",
    ]
    ok = all(check_module(name) for name in names)
    if not ok:
        print("\nInstall compatible prebuilt wheels from GitHub Releases and rerun this script.")
        return 1
    print("\nEDGS CUDA extension imports passed.")
    return 0


if __name__ == "__main__":
    raise SystemExit(main())