Make real guard-rails tests query the driver version.

This keeps the host-backed compatibility checks aligned with the actual driver state instead of relying on a fixed encoded version in the real-environment tests.

Made-with: Cursor

Author: rwgk

cuda_pathfinder/tests/local_helpers.py

@@ -12,6 +12,7 @@
 from cuda.pathfinder._headers.find_nvidia_headers import (
     locate_nvidia_header_directory as locate_nvidia_header_directory_raw,
 )
+from cuda.pathfinder._utils import driver_info
 from cuda.pathfinder._utils.toolkit_info import CudaToolkitVersion, read_cuda_header_version
 
 
@@ -68,3 +69,11 @@ def require_real_cuda_toolkit_version_from_cuda_h() -> LocatedRealCudaToolkitVer
         header_dir=located.abs_path,
         found_via=located.found_via,
     )
+
+
+def require_real_driver_cuda_version() -> driver_info.DriverCudaVersion:
+    """Return the real-host CUDA driver version or skip if it cannot be queried."""
+    try:
+        return driver_info.query_driver_cuda_version()
+    except driver_info.QueryDriverCudaVersionError as exc:
+        pytest.skip(f"Could not query the CUDA driver version for a real driver installation: {exc}")

cuda_pathfinder/tests/test_compatibility_guard_rails.py

@@ -27,10 +27,12 @@
     locate_nvidia_header_directory as locate_nvidia_header_directory_raw,
 )
 from cuda.pathfinder._utils.env_vars import get_cuda_path_or_home
+from cuda.pathfinder._utils import driver_info
 from cuda.pathfinder._utils.driver_info import DriverCudaVersion, QueryDriverCudaVersionError
 from cuda.pathfinder._utils.toolkit_info import read_cuda_header_version
 from local_helpers import (
     locate_real_cuda_toolkit_version_from_cuda_h,
+    require_real_driver_cuda_version,
     require_real_cuda_toolkit_version_from_cuda_h,
 )
 
@@ -46,18 +48,22 @@ def _default_process_wide_guard_rails_mode(monkeypatch):
 
 
 @pytest.fixture
-def clear_real_cuda_h_probe_caches():
+def clear_real_host_probe_caches():
     locate_real_cuda_toolkit_version_from_cuda_h.cache_clear()
     locate_nvidia_header_directory_raw.cache_clear()
     _resolve_system_loaded_abs_path_in_subprocess.cache_clear()
     get_cuda_path_or_home.cache_clear()
     read_cuda_header_version.cache_clear()
+    driver_info._load_nvidia_dynamic_lib.cache_clear()
+    driver_info.query_driver_cuda_version.cache_clear()
     yield
     locate_real_cuda_toolkit_version_from_cuda_h.cache_clear()
     locate_nvidia_header_directory_raw.cache_clear()
     _resolve_system_loaded_abs_path_in_subprocess.cache_clear()
     get_cuda_path_or_home.cache_clear()
     read_cuda_header_version.cache_clear()
+    driver_info._load_nvidia_dynamic_lib.cache_clear()
+    driver_info.query_driver_cuda_version.cache_clear()
 
 
 def _write_cuda_h(
@@ -678,17 +684,22 @@ def test_find_nvidia_header_directory_returns_none_when_unresolved(monkeypatch):
     assert guard_rails.find_nvidia_header_directory("nvrtc") is None
 
 
-@pytest.mark.usefixtures("clear_real_cuda_h_probe_caches")
+@pytest.mark.usefixtures("clear_real_host_probe_caches")
 def test_real_wheel_ctk_items_are_compatible(info_summary_append):
     real_ctk = require_real_cuda_toolkit_version_from_cuda_h()
+    real_driver = require_real_driver_cuda_version()
     info_summary_append(
         f"real cuda.h CTK version={real_ctk.version.major}.{real_ctk.version.minor} "
         f"via {real_ctk.found_via} at {real_ctk.cuda_h_path!r}"
     )
+    info_summary_append(
+        "real driver CUDA version="
+        f"{real_driver.major}.{real_driver.minor} (encoded={real_driver.encoded})"
+    )
     guard_rails = CompatibilityGuardRails(
         ctk_major=real_ctk.version.major,
         ctk_minor=real_ctk.version.minor,
-        driver_cuda_version=_driver_cuda_version(13000),
+        driver_cuda_version=real_driver,
     )
 
     try:
@@ -722,17 +733,22 @@ def test_real_wheel_ctk_items_are_compatible(info_summary_append):
         _assert_real_ctk_backed_path(path)
 
 
-@pytest.mark.usefixtures("clear_real_cuda_h_probe_caches")
+@pytest.mark.usefixtures("clear_real_host_probe_caches")
 def test_real_wheel_component_version_does_not_override_ctk_line(info_summary_append):
     real_ctk = require_real_cuda_toolkit_version_from_cuda_h()
+    real_driver = require_real_driver_cuda_version()
     info_summary_append(
         f"real cuda.h CTK version={real_ctk.version.major}.{real_ctk.version.minor} "
         f"via {real_ctk.found_via} at {real_ctk.cuda_h_path!r}"
     )
+    info_summary_append(
+        "real driver CUDA version="
+        f"{real_driver.major}.{real_driver.minor} (encoded={real_driver.encoded})"
+    )
     guard_rails = CompatibilityGuardRails(
         ctk_major=real_ctk.version.major,
         ctk_minor=real_ctk.version.minor,
-        driver_cuda_version=_driver_cuda_version(13000),
+        driver_cuda_version=real_driver,
     )
 
     try: