78 lines
2.6 KiB
JavaScript
78 lines
2.6 KiB
JavaScript
|
var vows = require('vows');
|
||
|
|
var assert = require('assert');
|
||
|
|
var suite = vows.describe('jStat.quantiles');
|
||
|
|
|
||
|
|
require('../env.js');
|
||
|
|
|
||
|
|
var tol = 0.0000001;
|
||
|
|
|
||
|
|
suite.addBatch({
|
||
|
|
'quantiles': {
|
||
|
|
'topic': function() {
|
||
|
|
return jStat;
|
||
|
|
},
|
||
|
|
'return basic quantiles': function(jStat) {
|
||
|
|
assert.deepEqual(jStat.quantiles([1, 2, 3, 4, 5, 6],
|
||
|
|
[0.25, 0.5, 0.75]),
|
||
|
|
[1.9375, 3.5, 5.0625]);
|
||
|
|
},
|
||
|
|
'quantiles from instance': function(jStat) {
|
||
|
|
assert.deepEqual(jStat([1, 2, 3, 4, 5, 6]).quantiles([0.25, 0.5, 0.75]),
|
||
|
|
[1.9375, 3.5, 5.0625]);
|
||
|
|
},
|
||
|
|
'quantiles matrix cols': function(jStat) {
|
||
|
|
assert.deepEqual(
|
||
|
|
jStat([[1, 2], [3, 4], [5, 6]]).quantiles([0.25, 0.5, 0.75]),
|
||
|
|
[[1.375, 3, 4.625], [2.375, 4, 5.625]]);
|
||
|
|
},
|
||
|
|
'quantiles normal dist': function(jStat) {
|
||
|
|
var arr = [-2.57313203, 9.84802638, 6.13625057, 8.41780777,
|
||
|
|
1.06749265, 2.1530631, 4.46082094, 8.26291053,
|
||
|
|
-9.28064583, 0.13434825];
|
||
|
|
|
||
|
|
var quantiles = [0.1, 0.3, 0.5, 0.8];
|
||
|
|
var results = jStat(arr).quantiles(quantiles);
|
||
|
|
var expected = [-6.59764031, 0.55426323, 3.30694202, 8.35197644];
|
||
|
|
var i;
|
||
|
|
|
||
|
|
for (i = 0; i < quantiles.length; i++) {
|
||
|
|
assert.epsilon(tol, results[i], expected[i]);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
'quantiles gamma dist': function(jStat) {
|
||
|
|
var arr = [6.20504472, 7.18983495, 6.29331634, 7.72493799,
|
||
|
|
6.44628893, 7.73877221, 8.26542627, 7.00870595,
|
||
|
|
6.72238426, 7.09363385, 6.60325838, 5.90180641,
|
||
|
|
5.79957376, 13.07687722, 6.65942804, 6.75392592,
|
||
|
|
6.41813748, 7.97086739, 9.36773336];
|
||
|
|
|
||
|
|
var quantiles = [0.83, 0.1, 0.3, 0.5, 0.8];
|
||
|
|
var results = jStat(arr).quantiles(quantiles, 0.4, 0.4);
|
||
|
|
var expected = [8.06983917, 5.99884267,
|
||
|
|
6.47140404, 6.75392592, 7.91516455];
|
||
|
|
|
||
|
|
for (var i=0; i < quantiles.length; i++)
|
||
|
|
assert.epsilon(tol, results[i], expected[i]);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
'#quantiles vector': {
|
||
|
|
'topic': function() {
|
||
|
|
jStat([1, 2, 3, 4, 5, 6]).quantiles([0.25, 0.5, 0.75], this.callback);
|
||
|
|
},
|
||
|
|
'quantiles callback': function(val, stat) {
|
||
|
|
assert.deepEqual(val, [1.9375, 3.5, 5.0625]);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
'#quantiles matrix cols': {
|
||
|
|
'topic': function() {
|
||
|
|
jStat([[1, 2], [3, 4], [5, 6]]).quantiles([0.25, 0.5, 0.75],
|
||
|
|
this.callback);
|
||
|
|
},
|
||
|
|
'quantiles matrix cols callback': function(val, stat) {
|
||
|
|
assert.deepEqual(val, [[1.375, 3, 4.625], [2.375, 4, 5.625]]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
suite.export(module);
|