/* ===========================================================
* JFreeChart : a free chart library for the Java(tm) platform
* ===========================================================
*
* (C) Copyright 2000-2011, by Object Refinery Limited and Contributors.
*
* Project Info: http://www.jfree.org/jfreechart/index.html
*
* This library is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or
* (at your option) any later version.
*
* This library is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
* License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
* USA.
*
* [Oracle and Java are registered trademarks of Oracle and/or its affiliates.
* Other names may be trademarks of their respective owners.]
*
* ------------------------
* TableXYDatasetTests.java
* ------------------------
* (C) Copyright 2003-2008, by Richard Atkinson and Contributors.
*
* Original Author: Richard Atkinson;
* Contributor(s): David Gilbert (for Object Refinery Limited);
*
* Changes
* -------
* 11-Aug-2003 : Version 1 (RA);
* 18-Aug-2003 : Added tests for event notification when removing and updating
* series (RA);
* 22-Sep-2003 : Changed to recognise that empty values are now null rather
* than zero (RA);
* 16-Feb-2004 : Added some additional tests (DG);
* 15-Jul-2004 : Switched getX() with getXValue() and getY() with
* getYValue() (DG);
* 02-Feb-2007 : Removed author tags all over JFreeChart sources (DG);
* 22-Apr-2008 : Added testPublicCloneable (DG);
*
*/
package org.jfree.data.xy.junit;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.ObjectInput;
import java.io.ObjectInputStream;
import java.io.ObjectOutput;
import java.io.ObjectOutputStream;
import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;
import org.jfree.data.xy.DefaultTableXYDataset;
import org.jfree.data.xy.XYSeries;
import org.jfree.util.PublicCloneable;
/**
* Tests for {@link DefaultTableXYDataset}.
*/
public class TableXYDatasetTests extends TestCase {
/**
* Returns the tests as a test suite.
*
* @return The test suite.
*/
public static Test suite() {
return new TestSuite(TableXYDatasetTests.class);
}
/**
* Constructs a new set of tests.
*
* @param name the name of the tests.
*/
public TableXYDatasetTests(String name) {
super(name);
}
/**
* Confirm that the equals method can distinguish all the required fields.
*/
public void testEquals() {
DefaultTableXYDataset d1 = new DefaultTableXYDataset();
DefaultTableXYDataset d2 = new DefaultTableXYDataset();
assertTrue(d1.equals(d2));
assertTrue(d2.equals(d1));
d1.addSeries(createSeries1());
assertFalse(d1.equals(d2));
d2.addSeries(createSeries1());
assertTrue(d1.equals(d2));
}
/**
* Confirm that cloning works.
*/
public void testCloning() {
DefaultTableXYDataset d1 = new DefaultTableXYDataset();
d1.addSeries(createSeries1());
DefaultTableXYDataset d2 = null;
try {
d2 = (DefaultTableXYDataset) d1.clone();
}
catch (CloneNotSupportedException e) {
System.err.println("Failed to clone.");
}
assertTrue(d1 != d2);
assertTrue(d1.getClass() == d2.getClass());
assertTrue(d1.equals(d2));
}
/**
* Verify that this class implements {@link PublicCloneable}.
*/
public void testPublicCloneable() {
DefaultTableXYDataset d1 = new DefaultTableXYDataset();
assertTrue(d1 instanceof PublicCloneable);
}
/**
* Serialize an instance, restore it, and check for equality.
*/
public void testSerialization() {
DefaultTableXYDataset d1 = new DefaultTableXYDataset();
d1.addSeries(createSeries2());
DefaultTableXYDataset d2 = null;
try {
ByteArrayOutputStream buffer = new ByteArrayOutputStream();
ObjectOutput out = new ObjectOutputStream(buffer);
out.writeObject(d1);
out.close();
ObjectInput in = new ObjectInputStream(
new ByteArrayInputStream(buffer.toByteArray())
);
d2 = (DefaultTableXYDataset) in.readObject();
in.close();
}
catch (Exception e) {
System.out.println(e.toString());
}
assertEquals(d1, d2);
}
/**
* Assorted tests.
*/
public void testTableXYDataset() {
XYSeries series1 = createSeries1();
XYSeries series2 = createSeries2();
DefaultTableXYDataset dataset = new DefaultTableXYDataset();
dataset.addSeries(series1);
dataset.addSeries(series2);
// Test that there are 6 X points and some specific values
assertEquals(6, dataset.getItemCount());
assertEquals(6, dataset.getX(0, 5).intValue());
assertEquals(null, dataset.getY(0, 5));
assertEquals(6, dataset.getX(1, 5).intValue());
assertEquals(2, dataset.getY(1, 5).intValue());
// after adding a point to a series, check that there are now 7
// items in each series
series2.add(7, 2);
assertEquals(7, dataset.getItemCount());
assertEquals(null, dataset.getY(0, 6));
assertEquals(2, dataset.getY(1, 6).intValue());
// Remove series 1
dataset.removeSeries(series1);
// Test that there are still 7 X points
assertEquals(7, dataset.getItemCount());
// Remove series 2 and add new series
dataset.removeSeries(series2);
series1 = createSeries1();
dataset.addSeries(series1);
// Test that there are now 4 X points
assertEquals(4, dataset.getItemCount());
}
/**
* A test for bug report 788597.
*/
public void test788597() {
DefaultTableXYDataset dataset = new DefaultTableXYDataset();
dataset.addSeries(createSeries1());
assertEquals(4, dataset.getItemCount());
dataset.removeAllSeries();
assertEquals(0, dataset.getItemCount());
}
/**
* Test that removing all values for a given x works.
*/
public void testRemoveAllValuesForX() {
DefaultTableXYDataset dataset = new DefaultTableXYDataset();
dataset.addSeries(createSeries1());
dataset.addSeries(createSeries2());
dataset.removeAllValuesForX(new Double(2.0));
assertEquals(5, dataset.getItemCount());
assertEquals(new Double(1.0), dataset.getX(0, 0));
assertEquals(new Double(3.0), dataset.getX(0, 1));
assertEquals(new Double(4.0), dataset.getX(0, 2));
assertEquals(new Double(5.0), dataset.getX(0, 3));
assertEquals(new Double(6.0), dataset.getX(0, 4));
}
/**
* Tests to see that pruning removes unwanted x values.
*/
public void testPrune() {
DefaultTableXYDataset dataset = new DefaultTableXYDataset();
dataset.addSeries(createSeries1());
dataset.addSeries(createSeries2());
dataset.removeSeries(1);
dataset.prune();
assertEquals(4, dataset.getItemCount());
}
/**
* Tests the auto-pruning feature.
*/
public void testAutoPrune() {
// WITH AUTOPRUNING
DefaultTableXYDataset dataset = new DefaultTableXYDataset(true);
dataset.addSeries(createSeriesA());
assertEquals(2, dataset.getItemCount()); // should be 2 items
dataset.addSeries(createSeriesB());
assertEquals(2, dataset.getItemCount()); // still 2
dataset.removeSeries(1);
assertEquals(1, dataset.getItemCount()); // 1 value pruned.
// WITHOUT AUTOPRUNING
DefaultTableXYDataset dataset2 = new DefaultTableXYDataset(true);
dataset2.addSeries(createSeriesA());
assertEquals(2, dataset2.getItemCount()); // should be 2 items
dataset2.addSeries(createSeriesB());
assertEquals(2, dataset2.getItemCount()); // still 2
dataset2.removeSeries(1);
assertEquals(1, dataset2.getItemCount()); // still 2.
}
/**
* Creates a series for testing.
*
* @return A series.
*/
private XYSeries createSeriesA() {
XYSeries s = new XYSeries("A", true, false);
s.add(1.0, 1.1);
s.add(2.0, null);
return s;
}
/**
* Creates a series for testing.
*
* @return A series.
*/
private XYSeries createSeriesB() {
XYSeries s = new XYSeries("B", true, false);
s.add(1.0, null);
s.add(2.0, 2.2);
return s;
}
/**
* Creates a series for testing.
*
* @return A series.
*/
private XYSeries createSeries1() {
XYSeries series1 = new XYSeries("Series 1", true, false);
series1.add(1.0, 1.0);
series1.add(2.0, 1.0);
series1.add(4.0, 1.0);
series1.add(5.0, 1.0);
return series1;
}
/**
* Creates a series for testing.
*
* @return A series.
*/
private XYSeries createSeries2() {
XYSeries series2 = new XYSeries("Series 2", true, false);
series2.add(2.0, 2.0);
series2.add(3.0, 2.0);
series2.add(4.0, 2.0);
series2.add(5.0, 2.0);
series2.add(6.0, 2.0);
return series2;
}
}