small fixes

Author: brandon-b-miller

numba_cuda/numba/cuda/tests/nrt/test_nrt.py

@@ -12,6 +12,7 @@
 from numba.np.arrayobj import (_check_const_str_dtype, is_nonelike,
                                ty_parse_dtype, ty_parse_shape, numpy_empty_nd)
 
+
 def cuda_empty(shape, dtype):
     pass
 
@@ -37,6 +38,7 @@ def impl(shape, dtype):
         msg = f"Cannot parse input types to function np.empty({shape}, {dtype})"
         raise errors.TypingError(msg)
 
+
 @unittest.skip
 class TestNrtRefCt(EnableNRTStatsMixin, TestCase):
 
@@ -49,19 +51,22 @@ def test_no_return(self):
         """
         Test issue #1291
         """
+        n = 10
 
         @cuda.jit
         def kernel():
-            for i in 10:
-                temp = np.zeros(2)
+            for i in range(n):
+                temp = np.zeros(2) # noqa: F841
             return 0
 
         init_stats = rtsys.get_allocation_stats()
         kernel[1,1]()
         cur_stats = rtsys.get_allocation_stats()
+        breakpoint()
         self.assertEqual(cur_stats.alloc - init_stats.alloc, n)
         self.assertEqual(cur_stats.free - init_stats.free, n)
 
+
 class TestNrtBasic(CUDATestCase):
     def test_nrt_launches(self):
         from pynvjitlink.patch import patch_numba_linker
@@ -102,5 +107,6 @@ def g(out_ary):
 
         self.assertEqual(out_ary[0], 1)
 
+
 if __name__ == '__main__':
     unittest.main()