address review comment + add missing API ref

leofang · leofang · commit c467526cdbe6 · 2025-07-23T09:21:20.000Z
diff --git a/cuda_bindings/docs/source/module/utils.rst b/cuda_bindings/docs/source/module/utils.rst
@@ -3,14 +3,15 @@
 
 .. module:: cuda.bindings.utils
 
-Utils module
-============
+utils
+=====
 
 Functions
 ---------
 
 .. autosummary::
    :toctree: generated/
 
+   get_cuda_native_handle
    get_minimal_required_cuda_ver_from_ptx_ver
    get_ptx_ver
diff --git a/cuda_bindings/tests/test_utils.py b/cuda_bindings/tests/test_utils.py
@@ -75,10 +75,15 @@ def test_get_handle(target):
     assert handle == ptr
 
 
-def test_get_handle_error():
-    obj = (1, 2, 3, 4)
-    with pytest.raises(TypeError) as e:
-        handle = get_cuda_native_handle(obj)
-    obj = 1234
+@pytest.mark.parametrize(
+    "target",
+    (
+        (1, 2, 3, 4),
+        [5, 6],
+        {},
+        None,
+    )
+)
+def test_get_handle_error(target):
     with pytest.raises(TypeError) as e:
-        handle = get_cuda_native_handle(obj)
+        handle = get_cuda_native_handle(target)
