def test_native_unroll():
""" Tests that unrolling functionality works. """
a = 0
for i in dace.unroll(range(2, 4)):
a += i * i
assert a == 13
def sometest(A):
for i in dace.unroll(range(n)):
something[i](A)
def tounroll(A: dace.float64[1]):
for val in dace.unroll(mygenerator()):
A += val
def tounroll(A: dace.float64[1]):
for i in dace.unroll(range(1, 4)):
A[0] += i * i
if i in (2, 3):
break
def program(A):
for i in dace.unroll(range(n)):
called(A)
