Java tutorial
/* =========================================================== * JFreeChart : a free chart library for the Java(tm) platform * =========================================================== * * (C) Copyright 2000-2012, 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.chart.util.PublicCloneable; import org.jfree.data.xy.DefaultTableXYDataset; import org.jfree.data.xy.XYSeries; /** * Tests for {@link DefaultTableXYDataset}. */ public class TableXYDatasetTest extends TestCase { /** * Returns the tests as a test suite. * * @return The test suite. */ public static Test suite() { return new TestSuite(TableXYDatasetTest.class); } /** * Constructs a new set of tests. * * @param name the name of the tests. */ public TableXYDatasetTest(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; } }