MovingAverageTest.java 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. /* ===========================================================
  2. * JFreeChart : a free chart library for the Java(tm) platform
  3. * ===========================================================
  4. *
  5. * (C) Copyright 2000-2013, by Object Refinery Limited and Contributors.
  6. *
  7. * Project Info: http://www.jfree.org/jfreechart/index.html
  8. *
  9. * This library is free software; you can redistribute it and/or modify it
  10. * under the terms of the GNU Lesser General Public License as published by
  11. * the Free Software Foundation; either version 2.1 of the License, or
  12. * (at your option) any later version.
  13. *
  14. * This library is distributed in the hope that it will be useful, but
  15. * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  16. * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
  17. * License for more details.
  18. *
  19. * You should have received a copy of the GNU Lesser General Public
  20. * License along with this library; if not, write to the Free Software
  21. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
  22. * USA.
  23. *
  24. * [Oracle and Java are registered trademarks of Oracle and/or its affiliates.
  25. * Other names may be trademarks of their respective owners.]
  26. *
  27. * ----------------------
  28. * MovingAverageTest.java
  29. * ----------------------
  30. * (C) Copyright 2003-2013, by Object Refinery Limited and Contributors.
  31. *
  32. * Original Author: David Gilbert (for Object Refinery Limited);
  33. * Contributor(s): -;
  34. *
  35. * Changes
  36. * -------
  37. * 14-Aug-2003 : Version 1 (DG);
  38. * 04-Oct-2004 : Eliminated NumberUtils usage (DG);
  39. *
  40. */
  41. package org.jfree.data.time;
  42. import static org.junit.Assert.assertEquals;
  43. import org.jfree.date.MonthConstants;
  44. import org.junit.Test;
  45. /**
  46. * Tests for the {@link MovingAverage} class.
  47. */
  48. public class MovingAverageTest {
  49. private static final double EPSILON = 0.0000000001;
  50. /**
  51. * A test for the values calculated from a time series.
  52. */
  53. @Test
  54. public void test1() {
  55. TimeSeries source = createDailyTimeSeries1();
  56. TimeSeries maverage = MovingAverage.createMovingAverage(source,
  57. "Moving Average", 3, 3);
  58. // the moving average series has 7 items, the first three
  59. // days (11, 12, 13 August are skipped)
  60. assertEquals(7, maverage.getItemCount());
  61. double value = maverage.getValue(0).doubleValue();
  62. assertEquals(14.1, value, EPSILON);
  63. value = maverage.getValue(1).doubleValue();
  64. assertEquals(13.4, value, EPSILON);
  65. value = maverage.getValue(2).doubleValue();
  66. assertEquals(14.433333333333, value, EPSILON);
  67. value = maverage.getValue(3).doubleValue();
  68. assertEquals(14.933333333333, value, EPSILON);
  69. value = maverage.getValue(4).doubleValue();
  70. assertEquals(19.8, value, EPSILON);
  71. value = maverage.getValue(5).doubleValue();
  72. assertEquals(15.25, value, EPSILON);
  73. value = maverage.getValue(6).doubleValue();
  74. assertEquals(12.5, value, EPSILON);
  75. }
  76. /**
  77. * Creates a sample series.
  78. *
  79. * @return A sample series.
  80. */
  81. private TimeSeries createDailyTimeSeries1() {
  82. TimeSeries series = new TimeSeries("Series 1", Day.class);
  83. series.add(new Day(11, MonthConstants.AUGUST, 2003), 11.2);
  84. series.add(new Day(13, MonthConstants.AUGUST, 2003), 13.8);
  85. series.add(new Day(17, MonthConstants.AUGUST, 2003), 14.1);
  86. series.add(new Day(18, MonthConstants.AUGUST, 2003), 12.7);
  87. series.add(new Day(19, MonthConstants.AUGUST, 2003), 16.5);
  88. series.add(new Day(20, MonthConstants.AUGUST, 2003), 15.6);
  89. series.add(new Day(25, MonthConstants.AUGUST, 2003), 19.8);
  90. series.add(new Day(27, MonthConstants.AUGUST, 2003), 10.7);
  91. series.add(new Day(28, MonthConstants.AUGUST, 2003), 14.3);
  92. return series;
  93. }
  94. }