test: replace cuda driver, device, and context setup with fixtures where relevant (#1293)

* test: replace cuda driver, device, and context setup with fixtures where relevant

* test: make ctx an autouse fixture

Author: cpcloud

cuda_bindings/pixi.lock

@@ -0,0 +1,28 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
+
+import cuda.bindings.driver as cuda
+import pytest
+
+
+@pytest.fixture(scope="module")
+def cuda_driver():
+    (err,) = cuda.cuInit(0)
+    assert err == cuda.CUresult.CUDA_SUCCESS
+
+
+@pytest.fixture(scope="module")
+def device(cuda_driver):
+    err, device = cuda.cuDeviceGet(0)
+    assert err == cuda.CUresult.CUDA_SUCCESS
+    return device
+
+
+@pytest.fixture(scope="module", autouse=True)
+def ctx(device):
+    # Construct context
+    err, ctx = cuda.cuCtxCreate(None, 0, device)
+    assert err == cuda.CUresult.CUDA_SUCCESS
+    yield ctx
+    (err,) = cuda.cuCtxDestroy(ctx)
+    assert err == cuda.CUresult.CUDA_SUCCESS