# Copyright (C) 2006 Douglas Gregor <doug.gregor -at- gmail.com>.

# Use, modification and distribution is subject to the Boost Software
# License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
# http://www.boost.org/LICENSE_1_0.txt)

# Test basic communication.

from __future__ import print_function
import mpi

def ring_test(comm, value, kind, root):
    next_peer = (comm.rank + 1) % comm.size;
    prior_peer = (comm.rank + comm.size - 1) % comm.size;
    
    if comm.rank == root:
        print ("Passing %s around a ring from root %d..." % (kind, root)),
        comm.send(next_peer, 0, value)
        (other_value, stat) = comm.recv(return_status = True)
        assert value == other_value
        assert stat.source == prior_peer
        assert stat.tag == 0
    else:
        msg = comm.probe()
        other_value = comm.recv(msg.source, msg.tag)
        assert value == other_value
        comm.send(next_peer, 0, other_value)

    comm.barrier()
    if comm.rank == root:
        print ("OK")
    pass

if mpi.world.size < 2:
    print ("ERROR: ring_test.py must be executed with more than one process")
    mpi.world.abort(-1);
    
ring_test(mpi.world, 17, 'integers', 0)
ring_test(mpi.world, 17, 'integers', 1)
ring_test(mpi.world, 'Hello, World!', 'string', 0)
ring_test(mpi.world, 'Hello, World!', 'string', 1)
ring_test(mpi.world, ['Hello', 'MPI', 'Python', 'World'], 'list of strings', 0)
ring_test(mpi.world, ['Hello', 'MPI', 'Python', 'World'], 'list of strings', 1)