root/library/tests/rv_test.cpp @ 584

#include <string>
#include <sstream>
#include "../bdm/base/bdmbase.h"
#include "UnitTest++.h"

using namespace bdm;

TEST ( test_rv ) {
        RV::clear_all();

        RV a = RV ( "{a_in_test_rv }", "3" );
        CHECK ( a.equal ( a ) );
        CHECK_EQUAL ( 1, a.length() );
        CHECK_EQUAL ( 3, a.size ( 0 ) );
        CHECK_EQUAL ( std::string ( "a_in_test_rv" ), a.name ( 0 ) );

        RV b = RV ( "{b_in_test_rv }", "2" );
        CHECK ( !b.equal ( a ) );
        CHECK_EQUAL ( 0, b.mint() );

        RV c = RV ( "{c_in_test_rv }" );
        CHECK_EQUAL ( 1, c.length() );
        CHECK_EQUAL ( 1, c.size ( 0 ) );

        RV trv = RV ( "{e_in_test_rv f_in_test_rv }", "1 2", "3 4" );
        CHECK_EQUAL ( 2, trv.length() );
        CHECK_EQUAL ( 3, trv.mint() );

        // add a and b
        RV ab = a;
        bool added = ab.add ( b );
        CHECK ( added );
        CHECK_EQUAL ( 2, ab.length() );
        CHECK_EQUAL ( 3, ab.size ( 0 ) );
        CHECK_EQUAL ( std::string ( "a_in_test_rv" ), ab.name ( 0 ) );
        CHECK_EQUAL ( 2, ab.size ( 1 ) );
        CHECK_EQUAL ( std::string ( "b_in_test_rv" ), ab.name ( 1 ) );

        std::stringstream abss;
        abss << ab;
        CHECK_EQUAL ( std::string ( "1(3)=a_in_test_rv_{0}; 2(2)=b_in_test_rv_{0}; " ), abss.str() );

        // concat a, b and c
        RV abc = concat ( ab, c );
        CHECK_EQUAL ( 3, abc.length() );
        std::stringstream abcss;
        abcss << abc;
        CHECK_EQUAL ( std::string ( "1(3)=a_in_test_rv_{0}; 2(2)=b_in_test_rv_{0}; 3(1)=c_in_test_rv_{0}; " ), abcss.str() );

        // structure of a, b, c
        str s = abc.tostr();
        int exp_ids[] = { 1, 1, 1, 2, 2, 3 };
        int exp_sz = sizeof ( exp_ids ) / sizeof ( exp_ids[0] );
        CHECK_EQUAL ( exp_sz, s.ids.size() );
        CHECK_EQUAL ( exp_sz, s.times.size() );
        for ( int i = 0; i < exp_sz; ++i ) {
                CHECK_EQUAL ( exp_ids[i], s.ids ( i ) );
                CHECK_EQUAL ( 0, s.times ( i ) );
        }

        RV slice = abc ( 1, 2 );
        CHECK_EQUAL ( 1, slice.length() );
        CHECK_EQUAL ( 3, slice.size ( 0 ) );
        CHECK_EQUAL ( std::string ( "a_in_test_rv" ), slice.name ( 0 ) );

        // find a in abc
        ivec f = a.findself ( abc );
        CHECK_EQUAL ( 1, f.length() );
        CHECK_EQUAL ( 0, f ( 0 ) );

        // find abc in a
        f = abc.findself ( a );
        int exp_indices[] = { 0, -1, -1 };
        CHECK_EQUAL ( 3, f.length() );
        for ( unsigned i = 0;
                i < sizeof ( exp_indices ) / sizeof ( exp_indices[0] );
                ++i ) {
                CHECK_EQUAL ( exp_indices[i], f ( i ) );
        }

        // subtract b from abc
        RV ac = abc.subt ( b );
        std::stringstream acss;
        acss << ac;
        CHECK_EQUAL ( std::string ( "1(3)=a_in_test_rv_{0}; 3(1)=c_in_test_rv_{0}; " ), acss.str() );

        // data index of ac in abc
        ivec di = ac.dataind ( abc );
        int exp_di[] = { 0, 1, 2, 5 };
        exp_sz = sizeof ( exp_di ) / sizeof ( exp_di[0] );
        CHECK_EQUAL ( exp_sz, di.size() );
        for ( int i = 0; i < exp_sz; ++i ) {
                CHECK_EQUAL ( exp_di[i], di ( i ) );
        }

        // subselect
        RV bc = abc ( ivec ( "1 2" ) );
        std::stringstream bcss;
        bcss << bc;
        CHECK_EQUAL ( std::string ( "2(2)=b_in_test_rv_{0}; 3(1)=c_in_test_rv_{0}; " ), bcss.str() );

#if 0
        // actually doesn't select, just reorders the variables -
        // wonder if that's correct...
        bc = abc ( 1, 2 );
        bcss << bc;
        CHECK_EQUAL ( std::string ( "2(2)=b_in_test_rv_{0}; 3(1)=c_in_test_rv_{0}; " ), bcss.str() );
#endif

        // Copy indices between ba and ab
        RV ba = b;
        ba.add ( a );

        ivec ai;
        ivec bi;
        ba.dataind ( ac, ai, bi );

        int exp_ai[] = { 2, 3, 4 };
        exp_sz = sizeof ( exp_ai ) / sizeof ( exp_ai[0] );
        CHECK_EQUAL ( exp_sz, ai.size() );
        for ( unsigned i = 0;
                i < sizeof ( exp_ai ) / sizeof ( exp_ai[0] );
                ++i ) {
                CHECK_EQUAL ( exp_ai[i], ai ( i ) );
        }

        int exp_bi[] = { 0, 1, 2 };
        exp_sz = sizeof ( exp_bi ) / sizeof ( exp_bi[0] );
        CHECK_EQUAL ( exp_sz, bi.size() );
        for ( unsigned i = 0;
                i < sizeof ( exp_bi ) / sizeof ( exp_bi[0] );
                ++i ) {
                CHECK_EQUAL ( exp_bi[i], bi ( i ) );
        }
        
        // check uniqueness
        RV join=a;
        join.add(b);
        RV tmp=a; tmp.t(1);
        join.add(tmp);
        tmp=b; tmp.t(-1);
        join.add(tmp);
        
        CHECK_EQUAL(unique(join._ids()), vec_2(a.id(0), b.id(0)));
}