Add a test

Author: Thirumalai-Shaktivel

integration_tests/lpython_decorator_02.py

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