diff options
Diffstat (limited to 'src/boost/libs/mpi/test/python/ring_test.py')
-rw-r--r-- | src/boost/libs/mpi/test/python/ring_test.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/src/boost/libs/mpi/test/python/ring_test.py b/src/boost/libs/mpi/test/python/ring_test.py new file mode 100644 index 00000000..3c8b5b92 --- /dev/null +++ b/src/boost/libs/mpi/test/python/ring_test.py @@ -0,0 +1,42 @@ +# 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. + +import boost.parallel.mpi as 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) |