gcc/testsuite/gfortran.dg/findloc_4.f90

   1 ! { dg-do run }
   2 ! Test findloc with dim argument.
   3
   4 program main
   5   implicit none
   6   real, dimension(2,2) :: a, b
   7   logical, dimension(2,2) :: lo
   8   a = reshape([1.,2.,3.,4.], shape(a))
   9   b = reshape([1.,1.,1.,1.], shape(b))
  10
  11   lo = .true.
  12
  13   if (any(findloc(b,value=1.,dim=1) /= [1,1])) stop 1
  14   if (any(findloc(b,value=1.,dim=2) /= [1,1])) stop 2
  15   if (any(findloc(b,value=1.,dim=1,back=.true.) /= [2,2])) stop 3
  16   if (any(findloc(b,value=1.,dim=2,back=.true.) /= [2,2])) stop 4
  17   if (any(findloc(b,value=1.,dim=1,mask=lo) /= [1,1])) stop 5
  18
  19   if (any(findloc(b,value=1.,dim=1,mask=lo,back=.true.) /= [2,2])) stop 6
  20   if (any(findloc(b,value=1.,dim=1,mask=.not. lo) /= [0,0])) stop 7
  21   lo(1,1) = .false.
  22   if (any(findloc(b,value=1.,dim=1,mask=lo) /= [2,1])) stop 8
  23   if (any(findloc(a,value=1.5,dim=2,back=.true.) /= [0,0])) stop 9
  24   if (any(findloc(a,value=1,dim=1,mask=lo) /= [0,0])) stop 10
  25 end program main