Add test for exp (**) in arg list

rjfarmer · Jan 18, 2024 · e307d58 · e307d58
1 parent f529cab
commit e307d58
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/tests/explicit_arrays.f90 b/tests/explicit_arrays.f90
@@ -242,4 +242,14 @@ function func_return_2d_int_arr() result(x)
 
 	end function func_return_2d_int_arr
 
+
+	subroutine func_exp_array_in(n,x)
+		integer, intent(in) :: n
+		integer, dimension(2*n,2**n) :: x
+
+		x = 5
+
+	end subroutine func_exp_array_in
+
+
 end module explicit_arrays
diff --git a/tests/explicit_arrays_test.py b/tests/explicit_arrays_test.py
@@ -315,3 +315,13 @@ def test_func_return_2d_int_arr(self):
         assert np.array_equal(
             res.result, np.array([1, 2, 3, 4, 5, 6]).reshape(3, 2, order="F")
         )
+
+    def test_func_exp_array_in(self):
+        n = 3
+        y = np.zeros((2 * n, 2**n), dtype=int)
+
+        res = x.func_exp_array_in(n, y)
+
+        result = np.zeros(np.shape(y), dtype=int)
+        result[:, :] = 5
+        assert np.array_equal(res.args["x"], result)