auto_tests/tests/rolling_average.js

   1 /**
   2  * @fileoverview Tests for rolling averages.
   3  *
   4  * @author danvk@google.com (Dan Vanderkam)
   5  */
   6 var rollingAverageTestCase = TestCase("rolling-average");
   7
   8 rollingAverageTestCase.prototype.setUp = function() {
   9   document.body.innerHTML = "<div id='graph'></div>";
  10 };
  11
  12 rollingAverageTestCase.prototype.tearDown = function() {
  13 };
  14
  15 rollingAverageTestCase.prototype.testRollingAverage = function() {
  16   var opts = {
  17     width: 480,
  18     height: 320,
  19     rollPeriod: 1,
  20     showRoller: true
  21   };
  22   var data = "X,Y\n" +
  23       "0,0\n" +
  24       "1,1\n" +
  25       "2,2\n" +
  26       "3,3\n"
  27   ;
  28
  29   var graph = document.getElementById("graph");
  30   var g = new Dygraph(graph, data, opts);
  31
  32   g.setSelection(0); assertEquals("0: Y: 0", Util.getLegend());
  33   g.setSelection(1); assertEquals("1: Y: 1", Util.getLegend());
  34   g.setSelection(2); assertEquals("2: Y: 2", Util.getLegend());
  35   g.setSelection(3); assertEquals("3: Y: 3", Util.getLegend());
  36   assertEquals(1, g.rollPeriod());
  37
  38   g.updateOptions({rollPeriod: 2});
  39   g.setSelection(0); assertEquals("0: Y: 0", Util.getLegend());
  40   g.setSelection(1); assertEquals("1: Y: 0.5", Util.getLegend());
  41   g.setSelection(2); assertEquals("2: Y: 1.5", Util.getLegend());
  42   g.setSelection(3); assertEquals("3: Y: 2.5", Util.getLegend());
  43   assertEquals(2, g.rollPeriod());
  44
  45   g.updateOptions({rollPeriod: 3});
  46   g.setSelection(0); assertEquals("0: Y: 0", Util.getLegend());
  47   g.setSelection(1); assertEquals("1: Y: 0.5", Util.getLegend());
  48   g.setSelection(2); assertEquals("2: Y: 1", Util.getLegend());
  49   g.setSelection(3); assertEquals("3: Y: 2", Util.getLegend());
  50   assertEquals(3, g.rollPeriod());
  51
  52   g.updateOptions({rollPeriod: 4});
  53   g.setSelection(0); assertEquals("0: Y: 0", Util.getLegend());
  54   g.setSelection(1); assertEquals("1: Y: 0.5", Util.getLegend());
  55   g.setSelection(2); assertEquals("2: Y: 1", Util.getLegend());
  56   g.setSelection(3); assertEquals("3: Y: 1.5", Util.getLegend());
  57   assertEquals(4, g.rollPeriod());
  58 };
  59
  60 rollingAverageTestCase.prototype.testRollBoxDoesntDisapper = function() {
  61   var opts = {
  62     showRoller: true
  63   };
  64   var data = "X,Y\n" +
  65       "0,0\n" +
  66       "1,1\n" +
  67       "2,2\n" +
  68       "3,3\n"
  69   ;
  70
  71   var graph = document.getElementById("graph");
  72   var g = new Dygraph(graph, data, opts);
  73
  74   var roll_box = graph.getElementsByTagName("input");
  75   assertEquals(1, roll_box.length);
  76   assertEquals("1", roll_box[0].value);
  77
  78   graph.style.width = "500px";
  79   g.resize();
  80   assertEquals(1, roll_box.length);
  81   assertEquals("1", roll_box[0].value);
  82 };
  83
  84 // Regression test for http://code.google.com/p/dygraphs/issues/detail?id=426
  85 rollingAverageTestCase.prototype.testRollShortFractions = function() {
  86   var opts = {
  87     customBars: true,
  88     labels: ['x', 'A']
  89   };
  90   var data1 = [ [1, 10, [1, 20]] ];
  91   var data2 = [ [1, 10, [1, 20]],
  92                 [2, 20, [1, 30]],
  93               ];
  94
  95   var graph = document.getElementById("graph");
  96   var g = new Dygraph(graph, data2, opts);
  97
  98   var rolled1 = g.dataHandler_.rollingAverage(data1, 1, g);
  99   var rolled2 = g.dataHandler_.rollingAverage(data2, 1, g);
 100
 101   assertEquals(rolled1[0], rolled2[0]);
 102 };
 103
 104 rollingAverageTestCase.prototype.testRollCustomBars = function() {
 105   var opts = {
 106     customBars: true,
 107     rollPeriod: 2,
 108     labels: ['x', 'A']
 109   };
 110   var data = [ [1, [1, 10, 20]],
 111                [2, [1, 20, 30]],
 112                [3, [1, 30, 40]],
 113                [4, [1, 40, 50]]
 114               ];
 115
 116   var graph = document.getElementById("graph");
 117   var g = new Dygraph(graph, data, opts);
 118   var rolled = this.getRolledData(g, data, 1, 2);
 119   assertEquals([1, 10, [1, 20]], rolled[0]);
 120   assertEquals([2, 15, [1, 25]], rolled[1]);
 121   assertEquals([3, 25, [1, 35]], rolled[2]);
 122   assertEquals([4, 35, [1, 45]], rolled[3]);
 123 };
 124
 125 rollingAverageTestCase.prototype.testRollErrorBars = function() {
 126   var opts = {
 127     errorBars: true,
 128     rollPeriod: 2,
 129     labels: ['x', 'A']
 130   };
 131   var data = [ [1, [10, 1]],
 132                [2, [20, 1]],
 133                [3, [30, 1]],
 134                [4, [40, 1]]
 135               ];
 136
 137   var graph = document.getElementById("graph");
 138   var g = new Dygraph(graph, data, opts);
 139   var rolled = this.getRolledData(g, data, 1, 2);
 140   assertEquals([1, 10, [8, 12]], rolled[0]);
 141
 142   // variance = sqrt( pow(error) * rollPeriod)
 143   var variance = Math.sqrt(2);
 144   for (var i=1;i<data.length;i++) {
 145     var value = data[i][1][0] - 5;
 146     assertEquals("unexpected rolled average", value, rolled[i][1]);
 147     assertEquals("unexpected rolled min", value - variance, rolled[i][2][0]);
 148     assertEquals("unexpected rolled max", value + variance, rolled[i][2][1]);
 149   }
 150 };
 151
 152 rollingAverageTestCase.prototype.testRollFractions = function() {
 153   var opts = {
 154     fractions: true,
 155     rollPeriod: 2,
 156     labels: ['x', 'A']
 157   };
 158   var data = [ [1, [1, 10]],
 159                [2, [2, 10]],
 160                [3, [3, 10]],
 161                [4, [4, 10]]
 162               ];
 163
 164   var graph = document.getElementById("graph");
 165   var g = new Dygraph(graph, data, opts);
 166   var rolled = this.getRolledData(g, data, 1, 2);
 167   assertEquals([1, 10], rolled[0]);
 168   assertEquals([2, 15], rolled[1]);
 169   assertEquals([3, 25], rolled[2]);
 170   assertEquals([4, 35], rolled[3]);
 171 };
 172
 173 rollingAverageTestCase.prototype.testRollFractionsBars = function() {
 174   var opts = {
 175     fractions: true,
 176     errorBars: true,
 177     wilsonInterval: false,
 178     rollPeriod: 2,
 179     labels: ['x', 'A']
 180   };
 181   var data = [ [1, [1, 10]],
 182                [2, [2, 10]],
 183                [3, [3, 10]],
 184                [4, [4, 10]]
 185               ];
 186
 187   var graph = document.getElementById("graph");
 188   var g = new Dygraph(graph, data, opts);
 189   var rolled = this.getRolledData(g, data, 1, 2);
 190
 191   // precalculated rounded values expected
 192   var values = [10, 15, 25, 35];
 193   var lows = [-9, -1, 6, 14];
 194   var highs = [29, 31, 44, 56];
 195
 196   for (var i=0;i<data.length;i++) {
 197     assertEquals("unexpected rolled average", values[i], Math.round(rolled[i][1]));
 198     assertEquals("unexpected rolled min", lows[i], Math.round(rolled[i][2][0]));
 199     assertEquals("unexpected rolled max", highs[i], Math.round(rolled[i][2][1]));
 200   }
 201 };
 202
 203 rollingAverageTestCase.prototype.testRollFractionsBarsWilson = function() {
 204   var opts = {
 205     fractions: true,
 206     errorBars: true,
 207     wilsonInterval: true,
 208     rollPeriod: 2,
 209     labels: ['x', 'A']
 210   };
 211   var data = [ [1, [1, 10]],
 212                [2, [2, 10]],
 213                [3, [3, 10]],
 214                [4, [4, 10]]
 215               ];
 216
 217   var graph = document.getElementById("graph");
 218   var g = new Dygraph(graph, data, opts);
 219   var rolled = this.getRolledData(g, data, 1, 2);
 220
 221   //precalculated rounded values expected
 222   var values = [10, 15, 25, 35];
 223   var lows = [2, 5, 11, 18];
 224   var highs = [41, 37, 47, 57];
 225
 226   for (var i=0;i<data.length;i++) {
 227     assertEquals("unexpected rolled average", values[i], Math.round(rolled[i][1]));
 228     assertEquals("unexpected rolled min", lows[i], Math.round(rolled[i][2][0]));
 229     assertEquals("unexpected rolled max", highs[i], Math.round(rolled[i][2][1]));
 230   }
 231 };
 232
 233 rollingAverageTestCase.prototype.getRolledData = function(g, data, seriesIdx, rollPeriod){
 234   var options = g.attributes_;
 235   return g.dataHandler_.rollingAverage(g.dataHandler_.extractSeries(data, seriesIdx, options), rollPeriod, options);
 236 };