test: fail checkpoint coverage on missing bindings

Ensure checkpoint tests distinguish missing required cuda.bindings symbols from genuinely unsupported environments.

Author: rwgk

cuda_bindings/tests/test_cuda.py

@@ -861,6 +861,26 @@ def test_cuCheckpointProcessGetState_failure():
     assert state is None
 
 
+@pytest.mark.skipif(not supportsCudaAPI("cuCheckpointProcessGetState"), reason="When API was introduced")
+def test_cuCheckpoint_required_bindings_present():
+    required_bindings = (
+        "cuCheckpointProcessCheckpoint",
+        "cuCheckpointProcessGetRestoreThreadId",
+        "cuCheckpointProcessGetState",
+        "cuCheckpointProcessLock",
+        "cuCheckpointProcessRestore",
+        "cuCheckpointProcessUnlock",
+        "CUcheckpointGpuPair",
+        "CUcheckpointLockArgs",
+        "CUprocessState",
+        "CUcheckpointRestoreArgs",
+    )
+
+    missing = [name for name in required_bindings if not hasattr(cuda, name)]
+
+    assert missing == []
+
+
 def test_private_function_pointer_inspector():
     from cuda.bindings._bindings.cydriver import _inspect_function_pointer
 

cuda_core/tests/test_checkpoint.py

@@ -33,8 +33,19 @@ def _checkpoint_available():
     try:
         checkpoint._get_driver()
         return True
-    except RuntimeError:
-        return False
+    except RuntimeError as exc:
+        if _checkpoint_unavailable_can_skip(str(exc)):
+            return False
+        raise
+
+
+def _checkpoint_unavailable_can_skip(message):
+    return message.startswith(
+        (
+            "CUDA checkpointing is not supported by the installed NVIDIA driver.",
+            "CUDA checkpointing requires cuda.bindings with CUDA checkpoint API support. Found cuda.bindings ",
+        )
+    )
 
 
 needs_checkpoint = pytest.mark.skipif(
@@ -384,6 +395,37 @@ def test_public_symbols(self):
         assert checkpoint.__all__ == ["Process"]
         assert not hasattr(checkpoint, "ProcessStateType")
 
+    def test_checkpoint_available_skips_unsupported_driver(self, monkeypatch):
+        def raise_unsupported_driver():
+            raise RuntimeError("CUDA checkpointing is not supported by the installed NVIDIA driver.")
+
+        monkeypatch.setattr(checkpoint, "_get_driver", raise_unsupported_driver)
+
+        assert not _checkpoint_available()
+
+    def test_checkpoint_available_skips_old_bindings(self, monkeypatch):
+        def raise_old_bindings():
+            raise RuntimeError(
+                "CUDA checkpointing requires cuda.bindings with CUDA checkpoint API support. "
+                "Found cuda.bindings 12.7.0."
+            )
+
+        monkeypatch.setattr(checkpoint, "_get_driver", raise_old_bindings)
+
+        assert not _checkpoint_available()
+
+    def test_checkpoint_available_fails_missing_required_bindings(self, monkeypatch):
+        def raise_missing_binding():
+            raise RuntimeError(
+                "CUDA checkpointing requires cuda.bindings with CUDA checkpoint API support. "
+                "Missing: CUcheckpointRestoreArgs"
+            )
+
+        monkeypatch.setattr(checkpoint, "_get_driver", raise_missing_binding)
+
+        with pytest.raises(RuntimeError, match="Missing: CUcheckpointRestoreArgs"):
+            _checkpoint_available()
+
     def test_pid_is_read_only(self):
         proc = checkpoint.Process(1)
         assert proc.pid == 1