test_init_comm_final.F90

Go to the documentation of this file.

! Copyright (c) 2024 The YAC Authors
!
! SPDX-License-Identifier: BSD-3-Clause
 
 
#include "test_macros.inc"
 
program test_init_comm_finalize
 
  use utest
  use yac
  use mpi
 
  implicit none
 
  integer  :: comp_id
  integer  :: npes, mype, couple_npes, comp_npes
  integer  :: ierror
  integer  :: couple_communicator, comp_communicator
 
  call start_test("yac_finit and yac_ffinalize")
 
  call mpi_init(ierror)
 
  call mpi_comm_rank(mpi_comm_world, mype, ierror)
  call mpi_comm_size(mpi_comm_world, npes, ierror)
  call test(npes == 6)
 
  if (mype < 5) then
 
    call mpi_comm_split(mpi_comm_world, 0, 0, couple_communicator, ierror)
 
    call mpi_comm_size(couple_communicator, couple_npes, ierror)
    call test(couple_npes == 5)
 
    call yac_finit_comm( couple_communicator)
 
    call yac_fdef_comp(merge('comp_a','comp_b', mype < 3), comp_id)
 
    call yac_fget_comp_comm(comp_id, comp_communicator)
    call mpi_comm_size(comp_communicator, comp_npes, ierror)
    call test(comp_npes == merge(3, 2, mype < 3))
    call mpi_comm_free(comp_communicator, ierror)
 
    call yac_fenddef()
 
    call yac_ffinalize();
 
  else
    call mpi_comm_split(mpi_comm_world, 1, 0, couple_communicator, ierror)
 
    call mpi_comm_size(couple_communicator, couple_npes, ierror)
    call test(couple_npes == 1)
  endif
 
  call mpi_comm_free(couple_communicator, ierror)
  call mpi_finalize(ierror)
  call stop_test
  call exit_tests
 
end program test_init_comm_finalize