const_04.py

from lpython import i32, i16, Const
from numpy import empty, int16

def sum_const_array(array: Const[i16[:]], size: i32) -> i16:
    i: i32
    array_sum: i16 = i16(0)
    for i in range(size):
        array_sum += array[i]
    return array_sum

def test_const_array():
    arr: i16[4] = empty(4, dtype=int16)
    i: i32
    for i in range(4):
        arr[i] = i16(i)
    assert sum_const_array(arr, 4) == i16(6)

test_const_array()