Register a test

Author: Thirumalai-Shaktivel

integration_tests/CMakeLists.txt

@@ -619,5 +619,6 @@ RUN(NAME intrinsics_01     LABELS cpython llvm NOFAST) # any
 
 # lpython decorator
 RUN(NAME lpython_decorator_01    LABELS cpython)
+RUN(NAME lpython_decorator_02    LABELS cpython)
 
 COMPILE(NAME import_order_01     LABELS cpython llvm c) # any

integration_tests/lpython_decorator_02.py

@@ -1,15 +1,19 @@
 from numpy import array
-from lpython import i32, f64, lpython
+from lpython import i32, f64, lpython, TypeVar
+
+n = TypeVar("n")
 
 @lpython
-def multiply(n: i32, x: f64[:]) -> f64[:]:
+def multiply(n: i32, x: f64[:]) -> f64[n]:
     i: i32
     for i in range(n):
         x[i] *= 5.0
     return x
 
 def test():
-    x: f64[3] = array([11.0, 12.0, 13.0, 14., 15.])
-    print(multiply(5, x))
+    size: i32 = 5
+    x: f64[5] = array([11.0, 12.0, 13.0, 14.0, 15.0])
+    y: f64[5] = (multiply(size, x))
+    assert y[2] == 65.
 
 test()