From 2cdfc9a45433a9114ababf9212caddc398aaab84 Mon Sep 17 00:00:00 2001 From: Graham Markall Date: Wed, 25 Jan 2023 22:19:40 +0000 Subject: [PATCH] Use nonzero data in tests of Issue #3158 --- numba/tests/test_record_dtype.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/numba/tests/test_record_dtype.py b/numba/tests/test_record_dtype.py index 44621dde0d7..21f67f466ed 100644 --- a/numba/tests/test_record_dtype.py +++ b/numba/tests/test_record_dtype.py @@ -1766,13 +1766,16 @@ def test_issue_1469_3(self): def test_issue_3158_1(self): # A nested array dtype. item = np.dtype([('some_field', np.int32)]) - items = np.dtype([('items', item, 10)]) + items = np.dtype([('items', item, 3)]) @njit def fn(x): return x[0] - arr = np.zeros((2,), items) + arr = np.asarray([([(0,), (1,), (2,)],), + ([(4,), (5,), (6,)],)], + dtype=items) + expected = fn.py_func(arr) actual = fn(arr) @@ -1788,7 +1791,9 @@ def test_issue_3158_2(self): def fn(arr): return arr[0] - arr = np.zeros(2, dtype3) + arr = np.asarray([(False, [[(0, 1), (2, 3)], [(4, 5), (6, 7)]]), + (True, [[(8, 9), (10, 11)], [(12, 13), (14, 15)]])], + dtype=dtype3) expected = fn.py_func(arr) actual = fn(arr)