GraphMLWriter_unit_tests.hpp

/*
 * GraphMLWriter_unit_tests.hpp
 *
 *  Created on: Dec 9, 2014
 *      Author: i-bird
 */

#ifndef GRAPHMLWRITER_UNIT_TESTS_HPP_
#define GRAPHMLWRITER_UNIT_TESTS_HPP_

#define GS_SIZE 8

#include "GraphMLWriter.hpp"
#include "Graph/CartesianGraphFactory.hpp"
#include "util/util.hpp"

BOOST_AUTO_TEST_SUITE( graphml_writer_test )


struct ne_cp
{
    typedef boost::fusion::vector<float,float,float,double,long int,int,std::string> type;

    type data;

    static const unsigned int x = 0;
    static const unsigned int y = 1;
    static const unsigned int z = 2;
    static const unsigned int double_num = 3;
    static const unsigned int long_num = 4;
    static const unsigned int integer = 5;
    static const unsigned int string = 6;
    static const unsigned int max_prop = 7;

    float & get_x() {return boost::fusion::at_c<x>(data);}
    float & get_y() {return boost::fusion::at_c<y>(data);}
    float & get_z() {return boost::fusion::at_c<z>(data);}
    double & get_dn()   {return boost::fusion::at_c<double_num>(data);}
    long int & get_ln() {return boost::fusion::at_c<long_num>(data);}
    int & get_i()   {return boost::fusion::at_c<integer>(data);}
    std::string & get_str() {return boost::fusion::at_c<string>(data);}

    struct attributes
    {
        static const std::string name[max_prop];
    };

    static inline bool noPointers()
    {
        return true;
    }

    typedef float s_type;
};

// Initialize the attributes strings array
const std::string ne_cp::attributes::name[] = {"x","y","z","double_num","long_num","integer","string"};

BOOST_AUTO_TEST_CASE( graphml_writer_use)
{
    Vcluster<> & v_cl = create_vcluster();

    if (v_cl.getProcessUnitID() != 0)
        return;

#ifdef OPENFPM_PDATA

    if (v_cl.rank() != 0) {return;}
    std::string c2 = std::string("openfpm_io/test_data/test_graph2_test.graphml");
    std::string c3 = std::string("openfpm_io/test_data/test_graph_test.graphml");

#else

    std::string c2 = std::string("test_data/test_graph2_test.graphml");
    std::string c3 = std::string("test_data/test_graph_test.graphml");

#endif

    Graph_CSR<ne_cp,ne_cp> g_csr2;

    // Add 4 vertex and connect

    struct ne_cp n1;
    n1.get_x() = 1.0;
    n1.get_y() = 2.0;
    n1.get_z() = 3.0;
    n1.get_dn() = 4.0;
    n1.get_ln() = 5.0;
    n1.get_i() = 6.0;
    n1.get_str() = std::string("test");
    g_csr2.addVertex(n1);
    n1.get_str() = std::string("tes2");
    g_csr2.addVertex(n1);
    n1.get_str() = std::string("test3");
    g_csr2.addVertex(n1);
    n1.get_str() = std::string("test4");
    g_csr2.addVertex(n1);
    n1.get_str() = std::string("test5");

    g_csr2.addEdge(0,1,n1);
    n1.get_str() = std::string("test6");
    g_csr2.addEdge(2,1,n1);
    n1.get_str() = std::string("test7");
    g_csr2.addEdge(3,1,n1);
    n1.get_str() = std::string("test8");
    g_csr2.addEdge(2,0,n1);
    n1.get_str() = std::string("test9");
    g_csr2.addEdge(3,2,n1);

    // Create a graph ML
    GraphMLWriter<Graph_CSR<ne_cp,ne_cp>> gv2(g_csr2);
    gv2.write("test_graph2.graphml");

    // check that match

    bool test = compare("test_graph2.graphml",c2);
    BOOST_REQUIRE_EQUAL(true,test);


    CartesianGraphFactory<3,Graph_CSR<ne_cp,ne_cp>> g_factory;

    // Cartesian grid
    size_t sz[] = {GS_SIZE,GS_SIZE,GS_SIZE};

    // Box
    Box<3,float> box({0.0,0.0,0.0},{1.0,1.0,1.0});

    // Boundary conditions, non periodic
    size_t bc[] = {NON_PERIODIC,NON_PERIODIC,NON_PERIODIC};

    Graph_CSR<ne_cp,ne_cp> g_csr = g_factory.construct<5,NO_VERTEX_ID,float,2,ne_cp::x,ne_cp::y,ne_cp::z>(sz,box,bc);

    // Create a graph ML
    GraphMLWriter<Graph_CSR<ne_cp,ne_cp>> gw(g_csr);
    gw.write("test_graph.graphml");


    // check that match
    test = compare("test_graph.graphml",c3);
    BOOST_REQUIRE_EQUAL(true,test);
}

BOOST_AUTO_TEST_SUITE_END()


#endif /* GRAPHMLWRITER_UNIT_TESTS_HPP_ */