Skip to content

Commit da74bd5

Browse files
brandonwillardtwiecki
authored andcommitted
Compare single elements directly in compare_numba_and_py
1 parent c07dfe4 commit da74bd5

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/link/numba/test_basic.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,6 @@ def set_test_value(x, v):
100100

101101

102102
def compare_shape_dtype(x, y):
103-
(x,) = x
104-
(y,) = y
105103
return x.shape == y.shape and x.dtype == y.dtype
106104

107105

@@ -246,7 +244,7 @@ def assert_fn(x, y):
246244
for j, p in zip(numba_res, py_res):
247245
assert_fn(j, p)
248246
else:
249-
assert_fn(numba_res, py_res)
247+
assert_fn(numba_res[0], py_res[0])
250248

251249
return numba_res
252250

0 commit comments

Comments
 (0)