| ! Program to test the eoshift intrinsic |
| program intrinsic_eoshift |
| integer, dimension(3, 3) :: a |
| integer, dimension(3, 3, 2) :: b |
| |
| ! Scalar shift and scalar bound. |
| a = reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| a = eoshift (a, 1, 99, 1) |
| if (any (a .ne. reshape ((/2, 3, 99, 5, 6, 99, 8, 9, 99/), (/3, 3/)))) & |
| call abort |
| |
| a = reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| a = eoshift (a, -2, dim = 2) |
| if (any (a .ne. reshape ((/0, 0, 0, 0, 0, 0, 1, 2, 3/), (/3, 3/)))) & |
| call abort |
| |
| ! Array shift and scalar bound. |
| a = reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| a = eoshift (a, (/1, 0, -1/), 99, 1) |
| if (any (a .ne. reshape ((/2, 3, 99, 4, 5, 6, 99, 7, 8/), (/3, 3/)))) & |
| call abort |
| |
| a = reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| a = eoshift (a, (/2, -2, 0/), dim = 2) |
| if (any (a .ne. reshape ((/7, 0, 3, 0, 0, 6, 0, 2, 9/), (/3, 3/)))) & |
| call abort |
| |
| ! Scalar shift and array bound. |
| a = reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| a = eoshift (a, 1, (/99, -1, 42/), 1) |
| if (any (a .ne. reshape ((/2, 3, 99, 5, 6, -1, 8, 9, 42/), (/3, 3/)))) & |
| call abort |
| |
| a = reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| a = eoshift (a, -2, (/99, -1, 42/), 2) |
| if (any (a .ne. reshape ((/99, -1, 42, 99, -1, 42, 1, 2, 3/), (/3, 3/)))) & |
| call abort |
| |
| ! Array shift and array bound. |
| a = reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| a = eoshift (a, (/1, 0, -1/), (/99, -1, 42/), 1) |
| if (any (a .ne. reshape ((/2, 3, 99, 4, 5, 6, 42, 7, 8/), (/3, 3/)))) & |
| call abort |
| |
| a = reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| a = eoshift (a, (/2, -2, 0/), (/99, -1, 42/), 2) |
| if (any (a .ne. reshape ((/7, -1, 3, 99, -1, 6, 99, 2, 9/), (/3, 3/)))) & |
| call abort |
| |
| ! Test arrays > rank 2 |
| b(:, :, 1) = reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| b(:, :, 2) = 10 + reshape ((/1, 2, 3, 4, 5, 6, 7, 8, 9/), (/3, 3/)) |
| b = eoshift (b, 1, 99, 1) |
| if (any (b(:, :, 1) .ne. reshape ((/2, 3, 99, 5, 6, 99, 8, 9, 99/), (/3, 3/)))) & |
| call abort |
| if (any (b(:, :, 2) .ne. reshape ((/12, 13, 99, 15, 16, 99, 18, 19, 99/), (/3, 3/)))) & |
| call abort |
| |
| ! TODO: Test array sections |
| end program |