Jatin K MalikandGitHub
26f90b6998
Minor on the stable line. Faster Whisper and Parakeet extras, Speech Model simple setup, Snap uinput, Flatpak bundles on the tag, and QA fixes from the release branch. Fixes tracked in the PR. Post-release follow-ups: #834.
122 lines
5.1 KiB
Python
122 lines
5.1 KiB
Python
"""Regression checks for installer CUDA diagnostics."""
|
|
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
INSTALLER = Path(__file__).resolve().parents[1] / "install.sh"
|
|
|
|
|
|
def _installer_source() -> str:
|
|
return INSTALLER.read_text(encoding="utf-8")
|
|
|
|
|
|
class InstallerCudaDiagnosticsTests(unittest.TestCase):
|
|
def test_installer_disables_user_site_packages(self) -> None:
|
|
"""Keep user-site packages out of the venv, activation script, and wrappers."""
|
|
source = _installer_source()
|
|
|
|
self.assertGreaterEqual(source.count("export PYTHONNOUSERSITE=1"), 4)
|
|
self.assertIn('getattr(site, "ENABLE_USER_SITE", False)', source)
|
|
|
|
def test_cuda_toolkit_validation_requires_complete_root(self) -> None:
|
|
"""CUDA roots must not be accepted when only a stale /usr/local/cuda exists."""
|
|
source = _installer_source()
|
|
|
|
self.assertIn("validate_cuda_toolkit_root()", source)
|
|
self.assertIn("$CUDA_ROOT/bin/nvcc", source)
|
|
self.assertIn("$CUDA_ROOT/include/cuda_runtime.h", source)
|
|
self.assertIn("$CUDA_ROOT/lib64/libcudart.so*", source)
|
|
self.assertIn("Ignoring incomplete CUDA toolkit root", source)
|
|
|
|
def test_cuda_build_passes_explicit_cmake_toolkit_and_architecture(self) -> None:
|
|
"""CUDA builds should steer CMake away from stale defaults."""
|
|
source = _installer_source()
|
|
|
|
self.assertIn("-DCUDAToolkit_ROOT=$CUDA_ROOT", source)
|
|
self.assertIn("-DCMAKE_CUDA_COMPILER=$CUDA_ROOT/bin/nvcc", source)
|
|
self.assertIn("-DCMAKE_CUDA_ARCHITECTURES=$CUDA_ARCHS", source)
|
|
self.assertIn("CUDA_CMAKE_ARGS=$(get_cuda_cmake_args", source)
|
|
self.assertIn("GGML_CUDA=1", source)
|
|
|
|
def test_backend_menu_does_not_promise_to_install_cuda_toolkit(self) -> None:
|
|
"""NVIDIA boxes used to say the CUDA toolkit would be installed.
|
|
|
|
install.sh never installs it. Vulkan is tried first; CUDA is a fallback
|
|
only when a complete toolkit is already on the machine.
|
|
"""
|
|
source = _installer_source()
|
|
self.assertNotIn("CUDA toolkit will be installed", source)
|
|
self.assertIn("NVIDIA GPU detected (Vulkan)", source)
|
|
self.assertIn('printf " │ • %-*s│\\n" 54 "$RECOMMENDED_REASON"', source)
|
|
|
|
def test_gpu_build_failures_print_pip_log_tail_before_cpu_fallback(self) -> None:
|
|
"""Backend failures should expose the real pip/CMake log."""
|
|
source = _installer_source()
|
|
|
|
self.assertIn("print_pip_log_tail()", source)
|
|
self.assertIn("Vulkan build failed; checking for NVIDIA GPU to try CUDA", source)
|
|
self.assertIn("CUDA build failed.", source)
|
|
self.assertIn(
|
|
"Continuing with CPU-only pywhispercpp; GPU acceleration is not active.",
|
|
source,
|
|
)
|
|
|
|
def test_backend_verification_checks_gpu_libraries_and_cuda_linkage(self) -> None:
|
|
"""Do not report GPU support unless native backend libraries exist."""
|
|
source = _installer_source()
|
|
|
|
self.assertIn("verify_pywhispercpp_backend_install()", source)
|
|
self.assertIn("libggml-vulkan.so", source)
|
|
self.assertIn("libggml-cuda.so", source)
|
|
self.assertIn("libcuda.so.1", source)
|
|
self.assertIn("libcuda-[^]]+\\.so", source)
|
|
|
|
def test_patchelf_remediation_is_attempted_for_bundled_libcuda(self) -> None:
|
|
"""Bundled libcuda should trigger patchelf relinking when available."""
|
|
source = _installer_source()
|
|
|
|
self.assertIn("patchelf", source)
|
|
self.assertIn("--replace-needed", source)
|
|
self.assertIn("readelf not found", source)
|
|
|
|
def test_installer_includes_xsel_for_wayland_clipboard_fallback(self) -> None:
|
|
"""Fresh installs should include xsel for ydotool's layout-safe paste path."""
|
|
source = _installer_source()
|
|
|
|
required_lines = [
|
|
"local APT_PACKAGES_UBUNTU=",
|
|
"local APT_PACKAGES_DEBIAN_BASE=",
|
|
"local DNF_PACKAGES=",
|
|
"local PACMAN_PACKAGES=",
|
|
"local ZYPPER_PACKAGES=",
|
|
"local EMERGE_PACKAGES=",
|
|
"local APK_PACKAGES=",
|
|
"local XBPS_PACKAGES=",
|
|
"local EOPKG_PACKAGES=",
|
|
]
|
|
|
|
for name in required_lines:
|
|
line = next(line for line in source.splitlines() if name in line)
|
|
self.assertIn("xclip", line)
|
|
self.assertIn("xsel", line)
|
|
self.assertIn("wl-clipboard", line)
|
|
|
|
def test_ubuntu_22_package_list_omits_util_linux_extra(self) -> None:
|
|
"""util-linux-extra is probed via apt-cache, not hardcoded in base package lists."""
|
|
source = _installer_source()
|
|
|
|
ubuntu_line = next(
|
|
line for line in source.splitlines() if "local APT_PACKAGES_UBUNTU=" in line
|
|
)
|
|
debian_13_line = next(
|
|
line for line in source.splitlines() if "local APT_PACKAGES_DEBIAN_13_PLUS=" in line
|
|
)
|
|
self.assertNotIn("util-linux-extra", ubuntu_line)
|
|
self.assertNotIn("util-linux-extra", debian_13_line)
|
|
self.assertIn("apt-cache show util-linux-extra", source)
|
|
self.assertIn('APT_PACKAGES="$APT_PACKAGES util-linux-extra"', source)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|