nac3_sca/nac3standalone/demo/src/ndarray.py

def consume_ndarray_1(n: ndarray[float, Literal[1]]):
    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():
    n = np_ndarray([1])
    consume_ndarray_1(n)

def test_ndarray_empty():
    n = np_empty([1])
    consume_ndarray_1(n)

def test_ndarray_zeros():
    n = np_zeros([1])
    consume_ndarray_1(n)

def test_ndarray_ones():
    n = np_ones([1])
    consume_ndarray_1(n)

def test_ndarray_full():
    n_float = np_full([1], 2.0)
    consume_ndarray_1(n_float)
    n_i32 = np_full([1], 2)
    consume_ndarray_i32_1(n_i32)

def test_ndarray_eye():
    n = np_eye(2)
    consume_ndarray_2(n)

def test_ndarray_identity():
    n = np_identity(2)
    consume_ndarray_2(n)

def run() -> int32:
    test_ndarray_ctor()
    test_ndarray_empty()
    test_ndarray_zeros()
    test_ndarray_ones()
    test_ndarray_full()
    test_ndarray_eye()
    test_ndarray_identity()

    return 0