TST: Define new test skip in conftest instead of copy-paste

Author: carterbox

cuda_bindings/tests/conftest.py

@@ -0,0 +1,8 @@
+import os
+
+import pytest
+
+skipif_compute_sanitizer_is_running = pytest.mark.skipif(
+    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
+    reason="The compute-sanitizer is running, and this test causes an API error.",
+)

cuda_bindings/tests/test_cuda.py

@@ -5,13 +5,13 @@
 # this software. Any use, reproduction, disclosure, or distribution of
 # this software and related documentation outside the terms of the EULA
 # is strictly prohibited.
-import os
 import platform
 import shutil
 import textwrap
 
 import numpy as np
 import pytest
+from conftest import skipif_compute_sanitizer_is_running
 
 import cuda.cuda as cuda
 import cuda.cudart as cudart
@@ -84,10 +84,7 @@ def test_cuda_memcpy():
     assert err == cuda.CUresult.CUDA_SUCCESS
 
 
-@pytest.mark.skipif(
-    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason="The compute-sanitizer is running, and this test intentionally causes an API error.",
-)
+@skipif_compute_sanitizer_is_running
 def test_cuda_array():
     (err,) = cuda.cuInit(0)
     assert err == cuda.CUresult.CUDA_SUCCESS
@@ -241,10 +238,7 @@ def test_cuda_uuid_list_access():
     assert err == cuda.CUresult.CUDA_SUCCESS
 
 
-@pytest.mark.skipif(
-    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason="FIXME: This test causes an API error.",
-)
+@skipif_compute_sanitizer_is_running
 def test_cuda_cuModuleLoadDataEx():
     (err,) = cuda.cuInit(0)
     assert err == cuda.CUresult.CUDA_SUCCESS
@@ -632,10 +626,7 @@ def test_cuda_coredump_attr():
     assert err == cuda.CUresult.CUDA_SUCCESS
 
 
-@pytest.mark.skipif(
-    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason="The compute-sanitizer is running, and this test intentionally causes an API error.",
-)
+@skipif_compute_sanitizer_is_running
 def test_get_error_name_and_string():
     (err,) = cuda.cuInit(0)
     assert err == cuda.CUresult.CUDA_SUCCESS
@@ -964,10 +955,7 @@ def test_CUmemDecompressParams_st():
     assert int(desc.dstActBytes) == 0
 
 
-@pytest.mark.skipif(
-    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason="The compute-sanitizer is running, and this test intentionally causes an API error.",
-)
+@skipif_compute_sanitizer_is_running
 def test_all_CUresult_codes():
     max_code = int(max(cuda.CUresult))
     # Smoke test. CUDA_ERROR_UNKNOWN = 999, but intentionally using literal value.
@@ -1000,30 +988,21 @@ def test_all_CUresult_codes():
     assert num_good >= 76  # CTK 11.0.3_450.51.06
 
 
-@pytest.mark.skipif(
-    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason="The compute-sanitizer is running, and this test intentionally causes an API error.",
-)
+@skipif_compute_sanitizer_is_running
 def test_cuKernelGetName_failure():
     err, name = cuda.cuKernelGetName(0)
     assert err == cuda.CUresult.CUDA_ERROR_INVALID_VALUE
     assert name is None
 
 
-@pytest.mark.skipif(
-    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason="The compute-sanitizer is running, and this test intentionally causes an API error.",
-)
+@skipif_compute_sanitizer_is_running
 def test_cuFuncGetName_failure():
     err, name = cuda.cuFuncGetName(0)
     assert err == cuda.CUresult.CUDA_ERROR_INVALID_VALUE
     assert name is None
 
 
-@pytest.mark.skipif(
-    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason="The compute-sanitizer is running, and this test intentionally causes an API error.",
-)
+@skipif_compute_sanitizer_is_running
 @pytest.mark.skipif(
     driverVersionLessThan(12080) or not supportsCudaAPI("cuCheckpointProcessGetState"),
     reason="When API was introduced",

cuda_bindings/tests/test_cudart.py

@@ -7,10 +7,10 @@
 # is strictly prohibited.
 import ctypes
 import math
-import os
 
 import numpy as np
 import pytest
+from conftest import skipif_compute_sanitizer_is_running
 
 import cuda.cuda as cuda
 import cuda.cudart as cudart
@@ -71,10 +71,7 @@ def test_cudart_memcpy():
     assertSuccess(err)
 
 
-@pytest.mark.skipif(
-    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason="The compute-sanitizer is running, and this test intentionally causes an API error.",
-)
+@skipif_compute_sanitizer_is_running
 def test_cudart_hostRegister():
     # Use hostRegister API to check for correct enum return values
     page_size = 80

cuda_core/tests/conftest.py

@@ -64,3 +64,9 @@ def clean_up_cffi_files():
             os.remove(f)
         except FileNotFoundError:
             pass  # noqa: SIM105
+
+
+skipif_compute_sanitizer_is_running = pytest.mark.skipif(
+    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
+    reason="The compute-sanitizer is running, and this test causes an API error.",
+)

cuda_core/tests/test_cuda_utils.py

@@ -1,9 +1,9 @@
 # Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
 #
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
-import os
 
 import pytest
+from conftest import skipif_compute_sanitizer_is_running
 
 from cuda.bindings import driver, runtime
 from cuda.core.experimental._utils import cuda_utils
@@ -41,13 +41,8 @@ def test_runtime_cuda_error_explanations_health():
         assert not extra_expl
 
 
-@pytest.mark.skipif(
-    os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason=(
-        "The compute-sanitizer is running, and this test causes an API error "
-        "when the driver is too old to know about all of the error codes."
-    ),
-)
+# this test causes an API error when the driver is too old to know about all of the error codes
+@skipif_compute_sanitizer_is_running
 def test_check_driver_error():
     num_unexpected = 0
     for error in driver.CUresult:

cuda_core/tests/test_linker.py

@@ -1,9 +1,9 @@
 # Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
 #
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
-import os
 
 import pytest
+from conftest import skipif_compute_sanitizer_is_running
 
 from cuda.core.experimental import Device, Linker, LinkerOptions, Program, ProgramOptions, _linker
 from cuda.core.experimental._module import ObjectCode
@@ -141,10 +141,8 @@ def test_linker_link_invalid_target_type(compile_ptx_functions):
         linker.link("invalid_target")
 
 
-@pytest.mark.skipif(
-    is_culink_backend and os.environ.get("CUDA_PYTHON_SANITIZER_RUNNING", "0") == "1",
-    reason="The compute-sanitizer is running, and this test causes an API error using the culink API.",
-)
+# this test causes an API error when using the culink API
+@skipif_compute_sanitizer_is_running
 def test_linker_get_error_log(compile_ptx_functions):
     options = LinkerOptions(arch=ARCH)