mathjs/test/function/statistics/median.test.js
2015-03-31 22:23:32 +02:00

92 lines
3.2 KiB
JavaScript

var assert = require('assert');
var approx = require('../../../tools/approx');
var math = require('../../../index');
var BigNumber = math.type.BigNumber;
var Complex = math.type.Complex;
var DenseMatrix = math.type.DenseMatrix;
var Unit = math.type.Unit;
var median = math.median;
describe('median', function() {
it('should return the median of an even number of numbers', function() {
assert.equal(median(3,1), 2);
assert.equal(median(1,3), 2);
approx.equal(median(1,3,5,2), 2.5);
assert.equal(median(0,0,0,0), 0);
});
it('should return the median of an odd number of numbers', function() {
assert.equal(median(0), 0);
assert.equal(median(5), 5);
approx.equal(median(1,3,5,2,-1), 2);
assert.equal(median(0,0,0), 0);
});
it('should return the median of an even number of new BigNumbers', function() {
assert.deepEqual(median(new BigNumber(1),new BigNumber(4),new BigNumber(5),new BigNumber(2)),
new BigNumber(3));
});
it('should return the median of an odd number of new BigNumbers', function() {
assert.deepEqual(median(new BigNumber(1),new BigNumber(4),new BigNumber(2)),
new BigNumber(2));
});
it('should return the median of an even number of booleans', function() {
assert.strictEqual(median(true, true, false, false), 0.5);
});
it('should return the median of an odd number of booleans', function() {
assert.strictEqual(median(true, true, false), 1);
});
it('should return the median from an array', function() {
assert.equal(median([1,3,5,2,-5]), 2);
});
it('should return the median of units', function() {
assert.deepEqual(median([new Unit(5,'mm'), new Unit(15,'mm'), new Unit(10,'mm')]), new Unit(10,'mm'));
assert.deepEqual(median([new Unit(5,'mm'), new Unit(30,'mm'), new Unit(20,'mm'), new Unit(10,'mm')]), new Unit(15,'mm'));
});
it('should return the median from an 1d matrix', function() {
assert.equal(median(new DenseMatrix([1,3,5,2,-5])), 2);
});
it('should return the median from a 2d array', function() {
approx.equal(median([
[ 1, 4, 7],
[ 3, 0, 5]
]), 3.5);
});
it('should return the median from a 2d matrix', function() {
approx.equal(median(new DenseMatrix([
[ 1, 4, 7],
[ 3, 0, 5]
])), 3.5);
});
it('should throw an error if called with invalid number of arguments', function() {
assert.throws(function() {median()});
assert.throws(function() {median([], 2, 3)});
});
it('should throw an error if called with not yet supported argument dim', function() {
assert.throws(function() {median([], 2)}, /not yet supported/);
});
it('should throw an error if called with unsupported type of arguments', function() {
assert.throws(function () {median('A', 'C', 'D', 'B')}, /TypeError: Unexpected type of argument/);
assert.throws(function () {median('A', 'C', 'B')}, /TypeError: Unexpected type of argument/);
assert.throws(function () {median(0, 'B')}, /TypeError: Unexpected type of argument/);
assert.throws(function () {median(new Complex(2,3), new Complex(-1,2))}, /TypeError: No ordering relation is defined for complex numbers/);
});
it('should throw an error if called with an empty array', function() {
assert.throws(function() {median([])});
});
});