Compare single elements directly in compare_numba_and_py

Author: brandonwillard

tests/link/numba/test_basic.py

@@ -127,8 +127,6 @@ def set_test_value(x, v):
 
 
 def compare_shape_dtype(x, y):
-    (x,) = x
-    (y,) = y
     return x.shape == y.shape and x.dtype == y.dtype
 
 
@@ -286,7 +284,7 @@ def assert_fn(x, y):
         for j, p in zip(numba_res, py_res):
             assert_fn(j, p)
     else:
-        assert_fn(numba_res, py_res)
+        assert_fn(numba_res[0], py_res[0])
 
     return pytensor_numba_fn, numba_res