Test cuda_core examples

Author: mdboom

cuda_core/pyproject.toml

@@ -56,7 +56,7 @@ cu12 = ["cuda-bindings[all]==12.*"]
 cu13 = ["cuda-bindings[all]==13.*"]
 
 [dependency-groups]
-test = ["cython>=3.2,<3.3", "setuptools", "pytest>=6.2.4", "pytest-randomly", "pytest-repeat", "pytest-rerunfailures"]
+test = ["cython>=3.2,<3.3", "setuptools", "pytest>=6.2.4", "pytest-randomly", "pytest-repeat", "pytest-rerunfailures", "cffi"]
 ml-dtypes = ["ml-dtypes>=0.5.4,<0.6.0"]
 test-cu12 = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cupy-cuda12x; python_version < '3.14'", "cuda-toolkit[cudart]==12.*"]  # runtime headers needed by CuPy
 test-cu13 = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cupy-cuda13x; python_version < '3.14'", "cuda-toolkit[cudart]==13.*"]  # runtime headers needed by CuPy

cuda_core/tests/example_tests/test_basic_examples.py

@@ -1,24 +1,33 @@
-# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 # If we have subcategories of examples in the future, this file can be split along those lines
 
 import glob
 import os
+import subprocess
+import sys
 
 import pytest
 
-from cuda.core import Device
-
-from .utils import run_example
+DISALLOW_LIST = {
+    "gl_interop_plasma.py",  # requires a display
+}
 
 samples_path = os.path.join(os.path.dirname(__file__), "..", "..", "examples")
-sample_files = glob.glob(samples_path + "**/*.py", recursive=True)
+sample_files = [
+    x
+    for x in (os.path.basename(x) for x in glob.glob(samples_path + "**/*.py", recursive=True))
+    if x not in DISALLOW_LIST
+]
 
 
 @pytest.mark.parametrize("example", sample_files)
 class TestExamples:
-    def test_example(self, example, deinit_cuda):
-        run_example(samples_path, example)
-        if Device().device_id != 0:
-            Device(0).set_current()
+    def test_example(self, example):
+        example_path = os.path.join(samples_path, example)
+        process = subprocess.run([sys.executable, example_path], capture_output=True)  # noqa: S603
+        if process.returncode != 0:
+            print(process.stdout.decode())
+            print(process.stderr.decode(), file=sys.stderr)
+            raise AssertionError(f"Example failed with return code {process.returncode} ({example})")