fix imports

Author: brandon-b-miller

numba_cuda/numba/cuda/tests/cudadrv/test_linker.py

@@ -1,5 +1,6 @@
 import numpy as np
 import warnings
+from numba.cuda.testing import unittest
 from numba.cuda.testing import (skip_on_cudasim, skip_if_cuda_includes_missing)
 from numba.cuda.testing import CUDATestCase, test_data_dir
 from numba.cuda.cudadrv.driver import (CudaAPIError, Linker,
@@ -310,3 +311,7 @@ def test_get_local_mem_per_specialized(self):
         local_mem_size = compiled_specialized.get_local_mem_per_thread()
         calc_size = np.dtype(np.float64).itemsize * LMEM_SIZE
         self.assertGreaterEqual(local_mem_size, calc_size)
+
+
+if __name__ == '__main__':
+    unittest.main()

numba_cuda/numba/cuda/tests/cudadrv/test_nvjitlink.py

@@ -4,14 +4,14 @@
 from numba.cuda.cudadrv.driver import PyNvJitLinker
 
 import itertools
-from pynvjitlink.api import NvJitLinkError
 from numba.cuda import get_current_device
 from numba import cuda
 from numba import config
 
 HAVE_PYNVJITLINK = False
 try:
     import pynvjitlink  # noqa: F401
+    from pynvjitlink.api import NvJitLinkError
 
     HAVE_PYNVJITLINK = True
 except ImportError: