program ESMF_VMSendrecvUTest
!------------------------------------------------------------------------------
#include "ESMF_Macros.inc"
!==============================================================================
!BOP
! !PROGRAM: ESMF_VMSendrecvUTest - Unit test for VM Sendrecv Function
!
! !DESCRIPTION:
!
! The code in this file drives the F90 VM SendRecv tests. The VM
! Sendrecv function is complex enough to require a separate test file.
! It runs on multiple processors.
!
!-----------------------------------------------------------------------------
! !USES:
use ESMF_TestMod ! test methods
use ESMF
implicit none
!------------------------------------------------------------------------------
! The following line turns the CVS identifier string into a printable variable.
character(*), parameter :: version = &
'$Id$'
!------------------------------------------------------------------------------
! cumulative result: count failures; no failures equals "all pass"
integer :: result = 0
! individual test failure message
character(ESMF_MAXSTR) :: failMsg
character(ESMF_MAXSTR) :: name
character(len=8) :: strvalue
! local variables
integer:: i, rc
type(ESMF_VM):: vm
integer:: localPet, petCount
integer:: count, src, dst
integer, allocatable:: localData(:),soln(:)
real(ESMF_KIND_R8), allocatable:: r8_localData(:),r8_soln(:)
real(ESMF_KIND_R4), allocatable:: r4_localData(:),r4_soln(:)
type(ESMF_logical), allocatable:: local_logical(:),logical_soln(:)
integer, allocatable:: i_recvData(:)
real(ESMF_KIND_R8), allocatable:: r8_recvData(:)
real(ESMF_KIND_R4), allocatable:: r4_recvData(:)
type(ESMF_logical), allocatable:: recv_logical(:)
integer :: isum
real(ESMF_KIND_R8) :: R8Sum
real(ESMF_KIND_R4) :: R4Sum
!------------------------------------------------------------------------------
! The unit tests are divided into Sanity and Exhaustive. The Sanity tests are
! always run. When the environment variable, EXHAUSTIVE, is set to ON then
! the EXHAUSTIVE and sanity tests both run. If the EXHAUSTIVE variable is set
! Special strings (Non-exhaustive and exhaustive) have been
! added to allow a script to count the number and types of unit tests.
!------------------------------------------------------------------------------
call ESMF_TestStart(ESMF_SRCLINE, rc=rc)
if (rc /= ESMF_SUCCESS) call ESMF_Finalize(endflag=ESMF_END_ABORT)
! Get count of PETs and which PET number we are
call ESMF_VMGetGlobal(vm, rc=rc)
call ESMF_VMGet(vm, localPet=localPet, petCount=petCount, rc=rc)
! Allocate localData
count = 2
!------------------------------------------------------------------------------
! IMPORTANT NOTE:
! The correct operation of this unit test in uni-process mode depends on
! sufficient internal buffering inside ESMF_VMSendRecv! There will be an
! implementation specific threshold for count above which this unit test will
! start to hang!
!------------------------------------------------------------------------------
allocate(localData(count))
allocate(r8_localData(count))
allocate(r4_localData(count))
allocate(local_logical(count))
allocate(i_recvData(count))
allocate(r8_recvData(count))
allocate(r4_recvData(count))
allocate(recv_logical(count))
! Allocate the solution arrays
Allocate(soln(count))
Allocate(r8_soln(count))
Allocate(r4_soln(count))
allocate(logical_soln(count))
!Assign values
do i=1,count
localData(i) = localPet*100+i
r4_localData(i) = real( localData(i) , ESMF_KIND_R4)
r8_localData(i) = real( localData(i) , ESMF_KIND_R8)
if (mod(localData(i)+localPet,2).eq.0) then
local_logical(i)= ESMF_TRUE
else
local_logical(i)= ESMF_FALSE
endif
end do
src = localPet - 1
if (src < 0) src = petCount - 1
dst = localPet + 1
if (dst > petCount -1) dst = 0
!The solution to test against is..
do i=1,count
soln(i) = src*100+i
r8_soln(i) = real( soln(i) , ESMF_KIND_R8)
r4_soln(i) = real( r8_soln(i))
if ( mod(soln(i)+src,2) .eq. 0 ) then
logical_soln(i)= ESMF_TRUE
else
logical_soln(i)= ESMF_FALSE
endif
end do
!Test with integer arguments
!===========================
!NEX_UTest
! Send local data to dst
write(failMsg, *) "Did not RETURN ESMF_SUCCESS"
write(name, *) "Sending local data Test"
call ESMF_VMSendRecv(vm, sendData=localData, sendCount=count, dstPet=dst, &
recvData=i_recvData, recvCount=count, srcPet=src, rc=rc)
call ESMF_Test((rc.eq.ESMF_SUCCESS), name, failMsg, result, ESMF_SRCLINE)
!------------------------------------------------------------------------
!NEX_UTest
! Verify localData after VM Receive
isum=0
write(failMsg, *) "Wrong Local Data"
write(name, *) "Verify local data after receive Test"
print *,localPet, " After rcv i_recvData is ", i_recvData(1), &
i_recvData(2),"( should be ",soln(1),soln(2)," )"
isum=isum+ (i_recvData(1) - soln(1)) + (i_recvData(2) - soln(2))
call ESMF_Test((isum.eq.0), name, failMsg, result, ESMF_SRCLINE)
!Test with REAL_KIND_R4 arguments
!================================
!NEX_UTest
! Send local data to dst
write(failMsg, *) "Did not RETURN ESMF_SUCCESS"
write(name, *) "Sending local data Test"
call ESMF_VMSendRecv(vm, sendData=r4_localData, sendCount=count, &
dstPet=dst, recvData=r4_recvData, recvCount=count, srcPet=src, rc=rc)
call ESMF_Test((rc.eq.ESMF_SUCCESS), name, failMsg, result, ESMF_SRCLINE)
!------------------------------------------------------------------------
!NEX_UTest
! Verify localData after VM Receive
R4Sum=0.
write(failMsg, *) "Wrong Local Data"
write(name, *) "Verify local data after receive Test"
print *,localPet, "After recv r4_recvData is ", r4_recvData(1), &
r4_recvData(2),"( should be ", r4_soln(1), r4_soln(2)," )"
R4Sum=(r4_recvData(1) - r4_soln(1)) + &
(r4_recvData(2) - r4_soln(2))
call ESMF_Test( (R4Sum .eq. 0.0), name, failMsg, result, ESMF_SRCLINE)
!Test with ESMF_KIND_R8 arguments
!================================
!NEX_UTest
! Send local data to dst
write(failMsg, *) "Did not RETURN ESMF_SUCCESS"
write(name, *) "Sending local data Test"
call ESMF_VMSendRecv(vm, sendData=r8_localData, sendCount=count, &
dstPet=dst, recvData=r8_recvData, recvCount=count,srcPet=src, rc=rc)
call ESMF_Test((rc.eq.ESMF_SUCCESS), name, failMsg, result, ESMF_SRCLINE)
!------------------------------------------------------------------------
!NEX_UTest
! Verify localData after VM Receive
R8Sum=0.
write(failMsg, *) "Wrong Local Data"
write(name, *) "Verify local data after receive Test"
print *,localPet, "After recv r8_recvData is ", r8_recvData(1), &
r8_recvData(2),"( Should be ", r8_soln(1), r8_soln(2)," )"
R8Sum=(r8_recvData(1) - r8_soln(1)) + &
(r8_recvData(2) - r8_soln(2))
call ESMF_Test( (R8Sum .eq. 0.0), name, failMsg, result, ESMF_SRCLINE)
!Test with logical arguments
!===========================
!NEX_UTest
! Send local data to dst
write(failMsg, *) "Did not RETURN ESMF_SUCCESS"
write(name, *) "Sending local data Test"
call ESMF_VMSendRecv(vm, sendData=local_logical, sendCount=count, &
dstPet=dst, recvData=recv_logical, recvCount=count, srcPet=src, rc=rc)
call ESMF_Test((rc.eq.ESMF_SUCCESS), name, failMsg, result, ESMF_SRCLINE)
!------------------------------------------------------------------------
!NEX_UTest
! Verify localData after VM Receive
ISum=0
write(failMsg, *) "Wrong Local Data"
write(name, *) "Verify local data after receive Test"
call ESMF_LogicalString(recv_logical(1), strvalue, rc)
print *, localPet, "After recv: Recv_Logical(1) is ", trim(strvalue)
call ESMF_LogicalString(recv_logical(2), strvalue, rc)
print *, localPet, "After recv: Recv_Logical(2) is ", trim(strvalue)
call ESMF_LogicalString(logical_soln(1), strvalue, rc)
print *, localPet, "After recv: Logical_soln(1) is ", trim(strvalue)
call ESMF_LogicalString(logical_soln(2), strvalue, rc)
print *, localPet, "After recv: logical_soln(2) is ", trim(strvalue)
do i=1,count
if (recv_logical(i).ne. logical_soln(i)) ISum= ISum + 1
end do
call ESMF_Test( (ISum .eq. 0), name, failMsg, result, ESMF_SRCLINE)
deallocate(localData)
deallocate(r8_localData)
deallocate(r4_localData)
deallocate(local_logical)
deallocate(i_recvData)
deallocate(r8_recvData)
deallocate(r4_recvData)
deallocate(recv_logical)
deallocate(soln)
deallocate(r8_soln)
deallocate(r4_soln)
deallocate(logical_soln)
call ESMF_TestEnd(ESMF_SRCLINE)
end program ESMF_VMSendRecvUTest