HDF5_writer_unit_tests.hpp

/*
 * HDF5_writer_unit_test.hpp
 *
 *  Created on: May 1, 2017
 *      Author: i-bird
 */

#ifndef OPENFPM_IO_SRC_HDF5_WR_HDF5_WRITER_UNIT_TESTS_HPP_
#define OPENFPM_IO_SRC_HDF5_WR_HDF5_WRITER_UNIT_TESTS_HPP_

#include "HDF5_wr.hpp"

#include "hdf5.h"

BOOST_AUTO_TEST_SUITE( vd_hdf5_chckpnt_rstrt_test )

// Dimensionality
const size_t dim = 3;

BOOST_AUTO_TEST_CASE( vector_dist_hdf5_save_test )
{
    openfpm::vector<Point<3,float>> vpos;
    openfpm::vector<aggregate<float[dim]>> vprp;

    // Put forces

    for (size_t i = 0 ; i < 1024 ; i++)
    {
        Point<3,float> p;

        p.get(0) = i;
        p.get(1) = i+13;
        p.get(2) = i+17;

        vpos.add(p);

        vprp.add();
        vprp.template get<0>(vprp.size()-1)[0] = p.get(0) + 100.0;
        vprp.template get<0>(vprp.size()-1)[1] = p.get(1) + 200.0;
        vprp.template get<0>(vprp.size()-1)[2] = p.get(2) + 300.0;
    }

    HDF5_writer<VECTOR_DIST> h5;

    // Save the vector
    h5.save("vector_dist.h5",vpos,vprp);

    HDF5_reader<VECTOR_DIST> h5r;

    openfpm::vector<Point<3,float>> vpos2;
    openfpm::vector<aggregate<float[dim]>> vprp2;

    size_t g_m = 0;
    h5r.load("vector_dist.h5",vpos2,vprp2,g_m);

    BOOST_REQUIRE_EQUAL(1024ul,vpos2.size());
    BOOST_REQUIRE_EQUAL(1024ul,vprp2.size());

    BOOST_REQUIRE_EQUAL(1024ul,g_m);

    // Check that vpos == vpos2 and vprp2 == vprp2

    bool check = true;
    for (size_t i = 0 ; i < vpos.size() ; i++)
    {
        check &= (vpos.get(i) == vpos2.get(i));
        check &= (vprp.get_o(i) == vprp2.get_o(i));
    }

    BOOST_REQUIRE_EQUAL(check,true);
}



BOOST_AUTO_TEST_CASE( vector_dist_hdf5_load_test )
{
    Vcluster & v_cl = create_vcluster();

    openfpm::vector<Point<3,float>> vpos;
    openfpm::vector<aggregate<float[dim]>> vprp;

    HDF5_reader<VECTOR_DIST> h5;

    size_t g_m = 0;

    // Load the vector
    h5.load("test_data/vector_dist_24.h5",vpos,vprp,g_m);


    // Check total number of particles
    size_t n_part = vpos.size();
    v_cl.sum(n_part);
    v_cl.execute();

    BOOST_REQUIRE_EQUAL(n_part,1024ul*24ul);

    BOOST_REQUIRE_EQUAL(vpos.size(),vprp.size());

    bool check = true;

    for (size_t i = 0 ; i < vpos.size() ; i++)
    {
        check &= (vprp.template get<0>(i)[0] == vpos.template get<0>(i)[0] + 100.0);
        check &= (vprp.template get<0>(i)[1] == vpos.template get<0>(i)[1] + 200.0);
        check &= (vprp.template get<0>(i)[2] == vpos.template get<0>(i)[2] + 300.0);
    }


}

BOOST_AUTO_TEST_SUITE_END()


#endif /* OPENFPM_IO_SRC_HDF5_WR_HDF5_WRITER_UNIT_TESTS_HPP_ */