update test_patch.py

rapidsai · Oct 29, 2024 · 2c362d0 · 2c362d0
1 parent 41ef0ac
commit 2c362d0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pynvjitlink/tests/test_patch.py b/pynvjitlink/tests/test_patch.py
@@ -5,7 +5,8 @@
 
 import pytest
 from numba import cuda
-from pynvjitlink import NvJitLinkError, patch
+from pynvjitlink import patch
+from pynvjitlink._nvjitlinklib import nvJitLinkError
 from pynvjitlink.patch import (
     PatchedLinker,
     _numba_version_ok,
@@ -48,7 +49,7 @@ def test_create_no_cc_error():
 def test_invalid_arch_error():
     # CC 0.0 is not a valid compute capability
     with pytest.raises(
-        NvJitLinkError, match="NVJITLINK_ERROR_UNRECOGNIZED_OPTION error"
+        nvJitLinkError, match="ERROR_UNRECOGNIZED_OPTION"
     ):
         PatchedLinker(cc=(0, 0))