1# Copyright (C) 2006 Douglas Gregor <doug.gregor -at- gmail.com>. 2 3# Use, modification and distribution is subject to the Boost Software 4# License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at 5# http://www.boost.org/LICENSE_1_0.txt) 6 7# Test scan() collective. 8 9from __future__ import print_function 10import mpi 11from generators import * 12 13def scan_test(comm, generator, kind, op, op_kind): 14 if comm.rank == 0: 15 print ("Prefix reduction to %s of %s..." % (op_kind, kind)), 16 my_value = generator(comm.rank) 17 result = mpi.scan(comm, my_value, op) 18 expected_result = generator(0); 19 for p in range(1, comm.rank+1): 20 expected_result = op(expected_result, generator(p)) 21 22 assert result == expected_result 23 if comm.rank == 0: 24 print ("OK.") 25 return 26 27scan_test(mpi.world, int_generator, "integers", lambda x,y:x + y, "sum") 28scan_test(mpi.world, int_generator, "integers", lambda x,y:x * y, "product") 29scan_test(mpi.world, string_generator, "strings", lambda x,y:x + y, "concatenation") 30scan_test(mpi.world, string_list_generator, "list of strings", lambda x,y:x + y, "concatenation") 31