diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-27 18:24:20 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-27 18:24:20 +0000 |
commit | 483eb2f56657e8e7f419ab1a4fab8dce9ade8609 (patch) | |
tree | e5d88d25d870d5dedacb6bbdbe2a966086a0a5cf /src/boost/libs/mpi/test/python/gather_test.py | |
parent | Initial commit. (diff) | |
download | ceph-upstream.tar.xz ceph-upstream.zip |
Adding upstream version 14.2.21.upstream/14.2.21upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/boost/libs/mpi/test/python/gather_test.py')
-rw-r--r-- | src/boost/libs/mpi/test/python/gather_test.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/src/boost/libs/mpi/test/python/gather_test.py b/src/boost/libs/mpi/test/python/gather_test.py new file mode 100644 index 00000000..d56b3a41 --- /dev/null +++ b/src/boost/libs/mpi/test/python/gather_test.py @@ -0,0 +1,32 @@ +# 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 gather() collective. + +import boost.parallel.mpi as mpi +from generators import * + +def gather_test(comm, generator, kind, root): + if comm.rank == root: + print ("Gathering %s to root %d..." % (kind, root)), + my_value = generator(comm.rank) + result = mpi.gather(comm, my_value, root) + if comm.rank == root: + for p in range(0, comm.size): + assert result[p] == generator(p) + print "OK." + else: + assert result == None + return + +gather_test(mpi.world, int_generator, "integers", 0) +gather_test(mpi.world, int_generator, "integers", 1) +gather_test(mpi.world, gps_generator, "GPS positions", 0) +gather_test(mpi.world, gps_generator, "GPS positions", 1) +gather_test(mpi.world, string_generator, "strings", 0) +gather_test(mpi.world, string_generator, "strings", 1) +gather_test(mpi.world, string_list_generator, "list of strings", 0) +gather_test(mpi.world, string_list_generator, "list of strings", 1) |