standalone: Update ndarray.py to output all elements in ndarrays

This commit is contained in:
David Mak 2024-03-07 13:08:45 +08:00
parent c3b122acfc
commit 2edc1de0b6
1 changed files with 20 additions and 12 deletions

View File

@ -9,12 +9,6 @@ def output_float64(x: float):
def consume_ndarray_1(n: ndarray[float, Literal[1]]): def consume_ndarray_1(n: ndarray[float, Literal[1]]):
pass pass
def consume_ndarray_i32_1(n: ndarray[int32, Literal[1]]):
pass
def consume_ndarray_2(n: ndarray[float, Literal[2]]):
pass
def test_ndarray_ctor(): def test_ndarray_ctor():
n: ndarray[float, Literal[1]] = np_ndarray([1]) n: ndarray[float, Literal[1]] = np_ndarray([1])
consume_ndarray_1(n) consume_ndarray_1(n)
@ -26,31 +20,29 @@ def test_ndarray_empty():
def test_ndarray_zeros(): def test_ndarray_zeros():
n: ndarray[float, 1] = np_zeros([1]) n: ndarray[float, 1] = np_zeros([1])
output_float64(n[0]) output_float64(n[0])
consume_ndarray_1(n)
def test_ndarray_ones(): def test_ndarray_ones():
n: ndarray[float, 1] = np_ones([1]) n: ndarray[float, 1] = np_ones([1])
output_float64(n[0]) output_float64(n[0])
consume_ndarray_1(n)
def test_ndarray_full(): def test_ndarray_full():
n_float: ndarray[float, 1] = np_full([1], 2.0) n_float: ndarray[float, 1] = np_full([1], 2.0)
output_float64(n_float[0]) output_float64(n_float[0])
consume_ndarray_1(n_float)
n_i32: ndarray[int32, 1] = np_full([1], 2) n_i32: ndarray[int32, 1] = np_full([1], 2)
output_int32(n_i32[0]) output_int32(n_i32[0])
consume_ndarray_i32_1(n_i32)
def test_ndarray_eye(): def test_ndarray_eye():
n: ndarray[float, 2] = np_eye(2) n: ndarray[float, 2] = np_eye(2)
n0: ndarray[float, 1] = n[0] n0: ndarray[float, 1] = n[0]
v: float = n0[0] v: float = n0[0]
output_float64(v) output_float64(v)
consume_ndarray_2(n)
def test_ndarray_identity(): def test_ndarray_identity():
n: ndarray[float, 2] = np_identity(2) n: ndarray[float, 2] = np_identity(2)
consume_ndarray_2(n) output_float64(n[0][0])
output_float64(n[0][1])
output_float64(n[1][0])
output_float64(n[1][1])
def test_ndarray_fill(): def test_ndarray_fill():
n: ndarray[float, 2] = np_empty([2, 2]) n: ndarray[float, 2] = np_empty([2, 2])
@ -60,6 +52,21 @@ def test_ndarray_fill():
output_float64(n[1][0]) output_float64(n[1][0])
output_float64(n[1][1]) output_float64(n[1][1])
def test_ndarray_copy():
x: ndarray[float, 2] = np_identity(2)
y = x.copy()
x.fill(0.0)
output_float64(x[0][0])
output_float64(x[0][1])
output_float64(x[1][0])
output_float64(x[1][1])
output_float64(y[0][0])
output_float64(y[0][1])
output_float64(y[1][0])
output_float64(y[1][1])
def run() -> int32: def run() -> int32:
test_ndarray_ctor() test_ndarray_ctor()
test_ndarray_empty() test_ndarray_empty()
@ -69,5 +76,6 @@ def run() -> int32:
test_ndarray_eye() test_ndarray_eye()
test_ndarray_identity() test_ndarray_identity()
test_ndarray_fill() test_ndarray_fill()
test_ndarray_copy()
return 0 return 0