time-to-botec/squiggle/node_modules/jstat/test/distribution/gamma-test.js
NunoSempere b6addc7f05 feat: add the node modules
Necessary in order to clearly see the squiggle hotwiring.
2022-12-03 12:44:49 +00:00

67 lines
2.2 KiB
JavaScript

var vows = require('vows');
var assert = require('assert');
var suite = vows.describe('jStat.distribution');
require('../env.js');
suite.addBatch({
'gamma pdf': {
'topic': function() {
return jStat;
},
'check instance and static pdf method': function (jStat) {
var shape = 5;
var scale = 1;
var gamma = jStat.gamma(shape, scale);
var xValues = [-1, 0, 1];
var x;
for (var i = 0; i < xValues.length; i++) {
x = xValues[i];
pStatic = jStat.gamma.pdf(x, shape, scale);
pInstance = gamma.pdf(x);
if (isNaN(pStatic)) {
assert(isNaN(pInstance));
} else {
assert(pStatic === pInstance,
'Gamma pdf evaluated at ' +
x + ' should be equal for instance and static methods.');
}
}
},
// Checked against R's dgamma(x, shape, rate = 1, scale = 1/rate, log = FALSE)
// options(digits=10)
// dgamma(2, 1, scale=1)
// dgamma(5, 10, scale=2)
// dgamma(18, 22, scale=0.8)
// dgamma(-5, 7, scale=10)
'check pdf': function(jStat) {
tol = 0.000001;
assert.epsilon(tol, jStat.gamma.pdf(2, 1, 1), 0.1353352832);
assert.epsilon(tol, jStat.gamma.pdf(5, 10, 2), 0.000431450369);
assert.epsilon(tol, jStat.gamma.pdf(18, 22, 0.8), 0.1029848021);
assert.epsilon(tol, jStat.gamma.pdf(-5, 7, 10), 0);
},
//Checked against R's pgamma(q, shape, rate = 1/scale)
//That is, jStat.gamma.cdf(5, 10, 2) == pgamma(5, 10, 1/2)
'check cdf': function(jStat) {
tol = 0.000001;
assert.epsilon(tol, jStat.gamma.cdf(2, 1, 1), 0.8646647);
assert.epsilon(tol, jStat.gamma.cdf(5, 10, 2), 0.0002773);
assert.epsilon(tol, jStat.gamma.cdf(18, 22, 0.8), 0.5701725);
assert.epsilon(tol, jStat.gamma.cdf(-1, 5, 5), 0);
},
//Checked against R's qgamma(p, shape, rate = 1/scale)
//That is, jStat.gamma.inv(0.86, 10, 2) == qgamma(0.86, 10, 1/2)
'check inv': function(jStat) {
tol = 0.00001;
assert.epsilon(tol, jStat.gamma.inv(0.86, 10, 2), 26.83397);
assert.epsilon(tol, jStat.gamma.inv(0.57, 22, 0.8), 17.99833);
}
},
});
suite.export(module);