Start of ctest for mpifx

CMakeLists.txt

@@ -27,6 +27,7 @@ include(GNUInstallDirs)
 
 add_subdirectory(lib)
 if(NOT BUILD_EXPORTED_TARGETS_ONLY)
+  enable_testing()
   add_subdirectory(test)
 endif()
 

test/CMakeLists.txt

@@ -1,16 +1,37 @@
-set(targets
+set(tested
   test_allgather
   test_allgatherv
+  )
+set(targets
   test_allreduce
   test_bcast
   test_comm_split
   test_gather
   test_gatherv
   test_reduce
   test_scatter
-  test_scatterv)
+  test_scatterv
+  )
+list(APPEND targets ${tested})
 
 foreach(target IN LISTS targets)
-  add_executable(${target} ${target}.f90)
+  add_executable(${target} ${target}.f90 testhelper.f90)
   target_link_libraries(${target} MpiFx)
 endforeach()
+
+foreach(target IN LISTS tested)
+  add_test(NAME ${target}
+    COMMAND ${MPIEXEC_EXECUTABLE}
+    ${MPIEXEC_NUMPROC_FLAG}
+    ${MPIEXEC_MAX_NUMPROCS}
+    ${MPIEXEC_PREFLAGS}
+    ${CMAKE_CURRENT_BINARY_DIR}/${target}
+    ${MPIEXEC_POSTFLAGS}
+    )
+  set_tests_properties(${target} PROPERTIES
+    PASS_REGULAR_EXPRESSION "TestPASSED"
+    )
+  set_tests_properties(${target} PROPERTIES
+    FAIL_REGULAR_EXPRESSION "TestFAILED"
+    )
+endforeach()

test/test_allgather.f90

@@ -1,5 +1,7 @@
+!> Test various patterns of allgather
 program test_allgather
   use libmpifx_module
+  use testhelper
   implicit none
 
   type(mpifx_comm) :: mycomm
@@ -8,6 +10,7 @@ program test_allgather
   integer, allocatable :: recv1(:), recv2(:,:)
   character(100) :: formstr
   character(*), parameter :: label = "(I2.2,'-',I3.3,'|',1X"
+  logical :: tPassed
 
   call mpifx_init()
   call mycomm%init()
@@ -16,12 +19,16 @@ program test_allgather
   send0 = mycomm%rank * 2
   allocate(recv1(1 * mycomm%size))
   recv1(:) = 0
-  write(*, label // ",A,1X,I0)") 1, mycomm%rank, &
-      & "Send0 buffer:", send0
+  write(*, label // ",A,1X,I0)") 1, mycomm%rank, "Send0 buffer:", send0
   call mpifx_allgather(mycomm, send0, recv1)
   write(formstr, "(A,I0,A)") "A,", size(recv1), "(1X,I0))"
-  write(*, label // formstr) 2, mycomm%rank, &
-      & "Recv1 buffer:", recv1(:)
+  write(*, label // formstr) 2, mycomm%rank, "Recv1 buffer:", recv1(:)
+  if (sum(recv1) /= mycomm%size * (mycomm%size-1)) then
+    tPassed = .false.
+  else
+    tPassed = .true.
+  end if
+  call testReturn(mycomm, tPassed)
   deallocate(recv1)
 
   ! I1 -> I1
@@ -30,25 +37,33 @@ program test_allgather
   recv1(:) = 0
   send1(:) = [ mycomm%rank, mycomm%rank + 1 ]
   write(formstr, "(A,I0,A)") "A,", size(send1), "(1X,I0))"
-  write(*, label // formstr) 3, mycomm%rank, &
-      & "Send1 buffer:", send1(:)
+  write(*, label // formstr) 3, mycomm%rank, "Send1 buffer:", send1(:)
   call mpifx_allgather(mycomm, send1, recv1)
   write(formstr, "(A,I0,A)") "A,", size(recv1), "(1X,I0))"
-  write(*, label // formstr) 4, mycomm%rank, &
-      & "Recv1 buffer:", recv1
+  write(*, label // formstr) 4, mycomm%rank, "Recv1 buffer:", recv1
+  if (sum(recv1) /= mycomm%size**2) then
+    tPassed = .false.
+  else
+    tPassed = .true.
+  end if
+  call testReturn(mycomm, tPassed)
 
   ! I1 -> I2
   allocate(recv2(size(send1), mycomm%size))
   recv2(:,:) = 0
   send1(:) = [ mycomm%rank, mycomm%rank + 1 ]
   write(formstr, "(A,I0,A)") "A,", size(send1), "(1X,I0))"
-  write(*, label // formstr) 5, mycomm%rank, &
-      & "Send1 buffer:", send1(:)
+  write(*, label // formstr) 5, mycomm%rank, "Send1 buffer:", send1(:)
   call mpifx_allgather(mycomm, send1, recv2)
   write(formstr, "(A,I0,A)") "A,", size(recv2), "(1X,I0))"
-  write(*, label // formstr) 6, mycomm%rank, &
-      & "Recv2 buffer:", recv2
-
+  write(*, label // formstr) 6, mycomm%rank, "Recv2 buffer:", recv2
+  if (sum(recv1) /= mycomm%size**2) then
+    tPassed = .false.
+  else
+    tPassed = .true.
+  end if
+  call testReturn(mycomm, tPassed)
+
   call mpifx_finalize()
-  
+
 end program test_allgather

test/test_allgatherv.f90

@@ -1,5 +1,7 @@
+!> Test various patterns of allgatherv
 program test_allgatherv
   use libmpifx_module
+  use testhelper
   implicit none
 
   type(mpifx_comm) :: mycomm
@@ -12,6 +14,8 @@ program test_allgatherv
   integer :: ii, nrecv, nCol
   character(100) :: formstr
   character(*), parameter :: label = "(I2.2,'-',I3.3,'|',1X"
+  logical :: tPassed
+  integer :: iCount
 
   call mpifx_init()
   call mycomm%init()
@@ -33,6 +37,16 @@ program test_allgatherv
   if (mycomm%rank == mycomm%size - 1) then
     write(*, *) "Recv1 buffer:", recv1
   end if
+  iCount = 0
+  do ii = 1, mycomm%size
+    iCount = iCount + ii**2
+  end do
+  if (nint(sum(recv1)) /= iCount) then
+    tPassed = .false.
+  else
+    tPassed = .true.
+  end if
+  call testReturn(mycomm, tPassed)
   deallocate(recvcounts)
   deallocate(recv1)
 

test/testhelper.f90

@@ -0,0 +1,36 @@
+!> Helper routines for testers
+module testhelper
+  use libmpifx_module, only : mpifx_comm, mpifx_barrier, mpifx_finalize
+  implicit none
+
+  private
+  public :: testReturn
+
+contains
+
+  !> Return expected labels for ctest
+  subroutine testReturn(mycomm, tPassed)
+
+    type(mpifx_comm), intent(in) :: mycomm
+
+    logical, intent(in) :: tPassed
+
+    call mpifx_barrier(mycomm)
+
+    if (tPassed) then
+      if (mycomm%rank == 0) then
+        ! label for ctest regex
+        write(*,*)'TestPASSED'
+      end if
+    else
+      if (mycomm%rank == 0) then
+        ! label for ctest regex
+        write(*,*)'TestFAILED'
+      end if
+      call mpifx_finalize()
+      stop
+    end if
+
+  end subroutine testReturn
+
+end module testhelper