root/library/tests/datalink_test.cpp @ 493

#include "../bdm/stat/exp_family.h"
#include "UnitTest++.h"

using namespace bdm;

TEST ( test_datalink ) {
        // RV names are global, and a, b and c are already taken by a
        // different test...
        RV a = RV ( "{dla }", "2" );
        RV b = RV ( "{dlb }" );
        RV c = RV ( "{dlc }" );

        RV ab = a;
        ab.add ( b );

        RV abc = ab;
        abc.add ( c );

        datalink dl ( ab, abc );
        vec total ( "0 37 42 66" );
        vec filtered = dl.pushdown ( total );
        int exp_f[] = { 0, 37, 42 };
        int exp_sz = sizeof ( exp_f ) / sizeof ( exp_f[0] );
        CHECK_EQUAL ( exp_sz, filtered.size() );
        for ( int i = 0; i < exp_sz; ++i ) {
                CHECK_EQUAL ( exp_f[i], filtered ( i ) );
        }
}

TEST ( test_datalink_m2e ) {
        RV a = RV ( "{dla }", "2" );
        RV b = RV ( "{dlb }" );
        RV c = RV ( "{dlc }" );

        RV ab = a;
        ab.add ( b );

        RV ba = a;
        ba.add ( b );

        RV abc = ab;
        abc.add ( c );

        datalink_m2e dl;
        dl.set_connection ( ba, c, abc );
        vec total ( "0 37 42 66" );
        vec cond = dl.get_cond ( total );
        CHECK_EQUAL ( 1, cond.size() );
        CHECK_EQUAL ( 66, cond ( 0 ) );
}

TEST ( test_datalink_m2m ) {
        RV a = RV ( "{dla }", "2" );
        RV b = RV ( "{dlb }" );
        RV c = RV ( "{dlc }" );

        datalink_m2m dl;
        dl.set_connection ( a, concat ( b, c ), concat ( a, b ), c );

        vec val ( "1 1.5 2" );
        vec cond ( "3" );

        vec p = dl.pushdown ( val );
        double exp_p[] = { 1.0, 1.5 };
        int exp_sz = sizeof ( exp_p ) / sizeof ( exp_p[0] );
        CHECK_EQUAL ( exp_sz, p.size() );
        for ( int i = 0; i < exp_sz; ++i ) {
                CHECK_EQUAL ( exp_p[i], p ( i ) );
        }

        vec dlcond = dl.get_cond ( val, cond );
        int exp_c[] = { 2, 3 };
        exp_sz = sizeof ( exp_c ) / sizeof ( exp_c[0] );
        CHECK_EQUAL ( exp_sz, dlcond.size() );
        for ( int i = 0; i < exp_sz; ++i ) {
                CHECK_EQUAL ( exp_c[i], dlcond ( i ) );
        }
}