root/sandbox/trunk/AdaBoost/test/org/xerial/mining/adaboost/BinaryDataSetTest.java

//--------------------------------------
// AdaBoost Project
//
// BinaryDataSetTest.java
// Since: Jul 9, 2007
//
// $URL$ 
// $Author$
//--------------------------------------
package org.xerial.mining.adaboost;


import java.io.IOException;

import static org.junit.Assert.*;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.xerial.util.FileResource;

public class BinaryDataSetTest
{

    @Before
    public void setUp() throws Exception
    {}

    @After
    public void tearDown() throws Exception
    {}

    
    @Test
    public void load() throws IOException
    {
        BinaryDataSet dataset = new BinaryDataSet();
        dataset.load(FileResource.find(BinaryDataSetTest.class, "sampledata.tab"));
        
        assertEquals(5, dataset.getColumnSize());
        assertEquals(16, dataset.getRowSize());
        
        assertEquals("T1", dataset.getColumnLabel(0));
        assertEquals("T2", dataset.getColumnLabel(1));
        assertEquals("T3", dataset.getColumnLabel(2));
        assertEquals("T4", dataset.getColumnLabel(3));
        assertEquals("objective", dataset.getColumnLabel(4));
        
        // the first data line
        assertEquals(true, dataset.get(0, 0));
        assertEquals(false, dataset.get(0, 1));
        assertEquals(true, dataset.get(0, 2));
        assertEquals(true, dataset.get(0, 3));

        // the last data line
        assertEquals(false, dataset.get(15, 0));
        assertEquals(false, dataset.get(15, 1));
        assertEquals(true, dataset.get(15, 2));
        assertEquals(false, dataset.get(15, 3));

    }
}