boost/libs/numeric/odeint/test_external/mpi/state_test.cpp

/*
 Copyright 2013 Karsten Ahnert
 Copyright 2013 Mario Mulansky
 Copyright 2013 Pascal Germroth

 Distributed under 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)
 */

#include <iostream>
#include <sstream>

#define BOOST_TEST_MODULE odeint_mpi
#include <boost/test/unit_test.hpp>

#include <boost/numeric/odeint/external/mpi/mpi.hpp>

using namespace boost::numeric::odeint;

boost::mpi::environment env;

BOOST_AUTO_TEST_SUITE( state_test_suite )

BOOST_AUTO_TEST_CASE( state_test )
{
    boost::mpi::communicator world;

    std::vector<size_t> in_data1, in_data2;
    mpi_state< std::vector<size_t> > state1(world), state2(world);

    // generate data on master
    if(world.rank() == 0) {
        in_data1.resize(31);
        in_data2.resize(33);
        for(size_t i = 0 ; i < in_data2.size() ; i++)
            in_data2[i] = i;
    }

    // copy to nodes
    split( in_data1, state1 );
    split( in_data2, state2 );

    {
        std::ostringstream ss;
        ss << "state[" << world.rank() << "] {"
           << state1().size() << ", "
           << state2().size() << "}\n";
        std::clog << ss.str() << std::flush;
    }

    // compare size
    BOOST_REQUIRE( !same_size( state1, state2 ) );

    // resize state1 to match state2.
    resize( state1, state2 );

    {
        std::ostringstream ss;
        ss << "state[" << world.rank() << "] 1:"
           << state1().size() << " 2:"
           << state2().size() << "\n";
        std::clog << ss.str() << std::flush;
    }

    // compare size
    BOOST_REQUIRE( same_size( state1, state2 ) );

    // copy state2 to state1
    copy( state2, state1 );

    BOOST_REQUIRE_EQUAL_COLLECTIONS(state1().begin(), state1().end(),
        state2().begin(), state2().end());
}


BOOST_AUTO_TEST_SUITE_END()