|| || || || || || || ||

---

TestRandomTree.h

Go to the documentation of this file.

#ifndef Impala_Core_Feature_Test_TestRandomTree_h
#define Impala_Core_Feature_Test_TestRandomTree_h

#include <cppunit/extensions/HelperMacros.h>
#include <algorithm>

#include "Core/Table/Equals.h"
#include "Core/Feature/RandomForest.h"

namespace Impala
{
namespace Core
{
namespace Feature
{

class TestRandomTree : public CPPUNIT_NS::TestFixture
{
    CPPUNIT_TEST_SUITE(TestRandomTree);
    CPPUNIT_TEST(testEquality);
    CPPUNIT_TEST(testGetCodeword);
    CPPUNIT_TEST(testGetSplit);
    CPPUNIT_TEST(testReadWrite);
    CPPUNIT_TEST(testTableConversion);
    CPPUNIT_TEST(testReadForest);
    CPPUNIT_TEST_SUITE_END();
    
public:

    void
    setUp()
    {
    }

    void
    tearDown()
    {
    }

    void 
    testEquality()
    {
        RandomTree t1(1);
        RandomTree t2(1);
        RandomTree t3(4);
        CPPUNIT_ASSERT(t1 == t2);
        CPPUNIT_ASSERT(t1 != t3);
        RandomTree t4(2, 4.1, new RandomTree(1), new RandomTree(2));
        RandomTree t5(2, 4.1, new RandomTree(1), new RandomTree(2));
        RandomTree t6(3, 4.1, new RandomTree(1), new RandomTree(2));
        RandomTree t7(2, 4.2, new RandomTree(1), new RandomTree(2));
        RandomTree t8(2, 4.1, new RandomTree(2), new RandomTree(2));
        RandomTree t9(2, 4.1, new RandomTree(1), new RandomTree(3));
        CPPUNIT_ASSERT(t4 == t5);
        CPPUNIT_ASSERT(t4 != t6);
        CPPUNIT_ASSERT(t4 != t7);
        CPPUNIT_ASSERT(t4 != t8);
        CPPUNIT_ASSERT(t4 != t9);
    }

    void 
    testGetCodeword()
    {
        RandomTree t(0, 1.2,
                     new RandomTree(1, 3.2, new RandomTree(1), new RandomTree(2)),
                     new RandomTree(1, 1.2, new RandomTree(3), new RandomTree(4)));
        Vector::VectorReal64 v1(0.7, 2.7);
        Vector::VectorReal64 v2(0.7, 4.7);
        Vector::VectorReal64 v3(1.7, 0.7);
        Vector::VectorReal64 v4(1.7, 2.7);
        CPPUNIT_ASSERT_EQUAL(1, t.GetCodeWord(v1));
        CPPUNIT_ASSERT_EQUAL(2, t.GetCodeWord(v2));
        CPPUNIT_ASSERT_EQUAL(3, t.GetCodeWord(v3));
        CPPUNIT_ASSERT_EQUAL(4, t.GetCodeWord(v4));
    }

    void
    testGetSplit()
    {
        RandomTree t1(3);
        int dim;
        double val;
        t1.GetSplit(dim, val);
        CPPUNIT_ASSERT_EQUAL(-1, dim);
        CPPUNIT_ASSERT_EQUAL(-1., val);
        RandomTree t2(2, 4., 0, 0);
        t2.GetSplit(dim, val);
        CPPUNIT_ASSERT_EQUAL(2, dim);
        CPPUNIT_ASSERT_EQUAL(4., val);
    }

    void 
    testReadWrite()
    {
        RandomTree t1(0, 1.2,
                      new RandomTree(1, 3.2, new RandomTree(1), new RandomTree(2)),
                      new RandomTree(1, 1.2, new RandomTree(3), new RandomTree(4)));
        RandomTree t2(8);
        RandomTreeTable table(0);
        Write(&t1, &table);
        CPPUNIT_ASSERT_EQUAL(7, table.Size());
        Write(&t2, &table);
        CPPUNIT_ASSERT_EQUAL(8, table.Size());

        int index=0;
        RandomTree* t1r = Read(&table, index);
        CPPUNIT_ASSERT(t1 == *t1r);
        CPPUNIT_ASSERT_EQUAL(7, index);
        RandomTree* t2r = Read(&table, index);
        CPPUNIT_ASSERT(t2 == *t2r);
        CPPUNIT_ASSERT_EQUAL(8, index);
        delete t1r;
        delete t2r;
    }

    void 
    testReadForest()
    {
        RandomTree t0(1);
        RandomTree t1(2);
        RandomTree t2(3);
        RandomTreeTable table(0);
        Write(&t0, &table);
        Write(&t1, &table);
        Write(&t2, &table);

        RandomForest f = ReadRandomForest(&table);
        CPPUNIT_ASSERT_EQUAL((UInt32) 3, (UInt32) f.size());
        CPPUNIT_ASSERT(t0 == *f[0]);
        CPPUNIT_ASSERT(t1 == *f[1]);
        CPPUNIT_ASSERT(t2 == *f[2]);
        CPPUNIT_ASSERT_NO_THROW(DeleteForest(f));
    }

    void 
    testTableConversion()
    {
        RandomTreeTable rtt(0);
        rtt.Add(-1, 2, 4.3);
        rtt.Add(0, -1, -1);
        rtt.Add(1, -1, -1);

        FeatureTable* ftt = MakeFeatureTable(&rtt);
        RandomTreeTable* rttest = MakeRandomTreeTable(ftt);
        CPPUNIT_ASSERT(Equals(&rtt, rttest));
        delete ftt;
        delete rttest;
    }

private:
    ILOG_CLASS;
};

ILOG_CLASS_INIT(TestRandomTree, Impala.Core.Feature);

CPPUNIT_TEST_SUITE_REGISTRATION( TestRandomTree );
    
} // namespace
} // namespace
} // namespace

#endif
    

---