Add fixture deinit_context_function

Use in test_module.py::test_num_args_error_handling
Add comments

Author: oleksandr-pavlyk

cuda_core/tests/conftest.py

@@ -51,6 +51,11 @@ def deinit_cuda():
     _device_unset_current()
 
 
+@pytest.fixture(scope="function")
+def deinit_context_function():
+    return _device_unset_current
+
+
 # samples relying on cffi could fail as the modules cannot be imported
 sys.path.append(os.getcwd())
 

cuda_core/tests/test_module.py

@@ -184,6 +184,7 @@ def test_saxpy_arguments(get_saxpy_kernel, cuda12_prerequisite_check):
 
     assert krn.num_arguments == 5
 
+    assert "ParamInfo" in str(type(krn).arguments_info.fget.__annotations__)
     arg_info = krn.arguments_info
     n_args = len(arg_info)
     assert n_args == krn.num_arguments
@@ -233,7 +234,7 @@ class ExpectedStruct(ctypes.Structure):
 
 
 @skipif_testing_with_compute_sanitizer
-def test_num_args_error_handling(deinit_cuda, cuda12_prerequisite_check):
+def test_num_args_error_handling(deinit_context_function, cuda12_prerequisite_check):
     if not cuda12_prerequisite_check:
         pytest.skip("Test requires CUDA 12")
     src = "__global__ void foo(int a) { }"
@@ -243,6 +244,10 @@ def test_num_args_error_handling(deinit_cuda, cuda12_prerequisite_check):
         name_expressions=("foo",),
     )
     krn = mod.get_kernel("foo")
+    # Unset current context using function from conftest
+    deinit_context_function()
+    # with no context, cuKernelGetParamInfo would report
+    # exception which we expect to handle by raising
     with pytest.raises(CUDAError):
         # assignment resolves linter error "B018: useless expression"
         _ = krn.num_arguments