M-Labs/nac3
M-Labs/nac3
8
1
Fork 12
Code Issues 99 Pull Requests 19 Releases Wiki Activity

standalone: Add test for multidim array index with one index

Browse Source
This commit is contained in:
David Mak 2024-06-20 11:46:34 +08:00
parent 2c4abcbd32
commit 5ad2db89d6
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nac3standalone/demo/src/ndarray.py
View File

@ -150,6 +150,15 @@ def test_ndarray_slices():
x2 = x[0::2, 0::2] x2 = x[0::2, 0::2]
output_ndarray_float_2(x2) output_ndarray_float_2(x2)
def test_ndarray_nd_idx():
x = np_identity(2)
x0: float = x[0, 0]
output_float64(x0)
output_float64(x[0, 1])
output_float64(x[1, 0])
output_float64(x[1, 1])
def test_ndarray_add(): def test_ndarray_add():
x = np_identity(2) x = np_identity(2)
y = x + np_ones([2, 2]) y = x + np_ones([2, 2])
@ -1393,6 +1402,7 @@ def run() -> int32:
test_ndarray_neg_idx() test_ndarray_neg_idx()
test_ndarray_slices() test_ndarray_slices()
test_ndarray_nd_idx()
test_ndarray_add() test_ndarray_add()
test_ndarray_add_broadcast() test_ndarray_add_broadcast()