[Tail-cvs] Tail/src/test/net/sf/tail/graphics SeriesDatasetTest.java, 1.1

[Márcio V. d. S. <mv...@us...>]

Update of /cvsroot/tail/Tail/src/test/net/sf/tail/graphics
In directory sc8-pr-cvs10.sourceforge.net:/tmp/cvs-serv20967/src/test/net/sf/tail/graphics

Added Files:
	SeriesDatasetTest.java 
Log Message:
Aperfeiçoado o gráfico e criado testes para o dataset


--- NEW FILE: SeriesDatasetTest.java ---
package net.sf.tail.graphics;


import static junit.framework.Assert.assertEquals;

import java.util.GregorianCalendar;
import java.util.LinkedList;
import java.util.List;

import net.sf.tail.DefaultTimeSeries;
import net.sf.tail.Indicator;
import net.sf.tail.Operation;
import net.sf.tail.OperationType;
import net.sf.tail.Tick;
import net.sf.tail.TimeSeries;
import net.sf.tail.Trade;
import net.sf.tail.sample.SampleIndicator;

import org.junit.Before;
import org.junit.Test;

public class SeriesDatasetTest {

	TimeSeries series; 
	List<Indicator<? extends Number>> indicators;
	List<Trade> trades;
	List<Tick> ticks;
	private SampleIndicator indicator1;
	private SampleIndicator indicator2;
	
	@Before
	public void setUp() throws Exception {

		ticks = new LinkedList<Tick>();
		ticks.add(new Tick(new GregorianCalendar(2007,6,6),1d));
		ticks.add(new Tick(new GregorianCalendar(2007,6,7),2d));
		ticks.add(new Tick(new GregorianCalendar(2007,6,8),3d));
		ticks.add(new Tick(new GregorianCalendar(2007,6,9),4d));
		
		
		series = new DefaultTimeSeries(ticks);
		indicators = new LinkedList<Indicator<? extends Number>>();
		trades = new LinkedList<Trade>();
		
		indicator1 = new SampleIndicator(new double[]{2d,3d,4d,5d});
		indicator2 = new SampleIndicator(new double[]{5d,4d,3d,2d});
		
		indicators.add(indicator1);
		indicators.add(indicator2);
		
		Operation entry = new Operation(0,OperationType.BUY);
		Operation exit = new Operation(1,OperationType.SELL);
		trades.add(new Trade(entry,exit));
		
		entry = new Operation(2,OperationType.BUY);
		exit = new Operation(3,OperationType.SELL);
		trades.add(new Trade(entry,exit));	
	}

	@Test
	public void testDefaultConstructor(){
		SeriesDataset dataset = new SeriesDataset(series,indicators,0,3);
		assertEquals(indicators.size(), dataset.getRowCount());
		assertEquals(series.getSize(), dataset.getColumnCount());
		assertEquals(4d, dataset.getValue(1, 1));
	}
	
	@Test
	public void testTradesConstructor(){
		SeriesDataset dataset = new SeriesDataset(series,indicators,0,3,trades);
		assertEquals(indicators.size()+trades.size(), dataset.getRowCount());
		assertEquals(series.getSize(), dataset.getColumnCount());
		assertEquals(2d, dataset.getValue(3, 1));
	}

	@Test
	public void testMoveRightWhenIndicatorDontHaveMoreDataUnmappedShouldDoNothing(){
		SeriesDataset dataset = new SeriesDataset(series,indicators,0,3,trades);
		dataset.moveRight();
		assertEquals(indicators.size()+trades.size(), dataset.getRowCount());
		assertEquals(series.getSize(), dataset.getColumnCount());
		assertEquals(2d, dataset.getValue(3, 1));
	}

	@Test
	public void testMoveLeftWhenIndicatorDontHaveMoreDataUnmappedShouldDoNothing(){
		SeriesDataset dataset = new SeriesDataset(series,indicators,0,3,trades);
		dataset.moveLeft();
		assertEquals(indicators.size()+trades.size(), dataset.getRowCount());
		assertEquals(series.getSize(), dataset.getColumnCount());
		assertEquals(2d, dataset.getValue(3, 1));
	}
	
	@Test
	public void testMoveRight(){
		SeriesDataset dataset = new SeriesDataset(series,indicators,0,3,trades);
		ticks.add(new Tick(new GregorianCalendar(2007,6,10),4d));
		indicator1.addValue(15d);
		indicator2.addValue(30d);
		
		dataset.moveRight();
		assertEquals(indicators.size()+trades.size(), dataset.getRowCount());
		assertEquals(series.getSize() - 1, dataset.getColumnCount());
		assertEquals(3d, dataset.getValue(1, 1));
	}
	
	@Test
	public void testMoveLeft(){
		SeriesDataset dataset = new SeriesDataset(series,indicators,0,3,trades);
		ticks.add(new Tick(new GregorianCalendar(2007,6,10),4d));
		indicator1.addValue(15d);
		indicator2.addValue(30d);
		
		dataset.moveRight();
		dataset.moveLeft();
		assertEquals(indicators.size()+trades.size(), dataset.getRowCount());
		assertEquals(series.getSize() - 1, dataset.getColumnCount());
		assertEquals(4d, dataset.getValue(1, 1));
	}
}