mirror of
https://github.com/josdejong/mathjs.git
synced 2025-12-08 19:46:04 +00:00
102 lines
3.2 KiB
JavaScript
102 lines
3.2 KiB
JavaScript
// test import
|
|
var assert = require('assert'),
|
|
math = require('../../../index')(),
|
|
approx = require('../../../tools/approx');
|
|
|
|
describe('import', function() {
|
|
|
|
beforeEach(function() {
|
|
math.import({
|
|
myvalue: 42,
|
|
hello: function (name) {
|
|
return 'hello, ' + name + '!';
|
|
}
|
|
}, {override: true});
|
|
});
|
|
|
|
afterEach(function() {
|
|
//TODO forget the members added in beforeEach
|
|
});
|
|
|
|
it('should import a custom member', function() {
|
|
assert.equal(math.myvalue * 2, 84);
|
|
assert.equal(math.hello('user'), 'hello, user!');
|
|
});
|
|
|
|
it('should not override existing functions', function() {
|
|
math.import({myvalue: 10});
|
|
approx.equal(math.myvalue, 42);
|
|
});
|
|
|
|
it('should override existing functions if forced', function() {
|
|
math.import({myvalue: 10}, {override: true});
|
|
approx.equal(math.myvalue, 10);
|
|
});
|
|
|
|
it('should parse the user defined members', function() {
|
|
if (math.parser) {
|
|
var parser = math.parser();
|
|
math.add(math.myvalue, 10)
|
|
parser.eval('myvalue + 10'); // 52
|
|
parser.eval('hello("user")'); // 'hello, user!'
|
|
}
|
|
});
|
|
|
|
var getSize = function (array) { return array.length; };
|
|
|
|
it('should wrap the custom functions automatically', function () {
|
|
math.import({ getSizeWrapped: getSize });
|
|
assert.equal(math.getSizeWrapped([1,2,3]), 3);
|
|
assert.equal(math.getSizeWrapped(math.matrix([1,2,3])), 3);
|
|
});
|
|
|
|
it('shouldn\'t wrap the custom functions if wrap = false', function () {
|
|
math.import({ getSizeNotWrapped: getSize }, { wrap: false });
|
|
assert.equal(math.getSizeNotWrapped([1,2,3]), 3);
|
|
assert.equal(math.getSizeNotWrapped(math.matrix([1,2,3])), undefined);
|
|
});
|
|
|
|
it('should extend math with numbers', function() {
|
|
// extend math.js with numbers.js
|
|
// examples copied from https://github.com/sjkaliski/numbers.js/blob/master/examples/statistic.js
|
|
math.import('numbers');
|
|
|
|
assert.equal(math.fibonacci(7), 13);
|
|
|
|
// Consider a data representing total follower count of a
|
|
// variety of users.
|
|
var followers = math.matrix([100, 50, 1000, 39, 283, 634, 3, 6123]);
|
|
|
|
// We can generate a report of summary statistics
|
|
// which includes the mean, 1st and 3rd quartiles,
|
|
// and standard deviation.
|
|
var report = math.report(followers);
|
|
approx.deepEqual(report, {
|
|
mean: 1029,
|
|
firstQuartile: 44.5,
|
|
median: 191.5,
|
|
thirdQuartile: 817,
|
|
standardDev: 1953.0897316815733
|
|
});
|
|
|
|
// Maybe we decide to become a bit more curious about
|
|
// trends in follower count, so we start conjecturing about
|
|
// our ability to "predict" trends.
|
|
// Let's consider the number of tweets those users have.
|
|
var tweets = math.matrix([100, 10, 400, 5, 123, 24, 302, 2000]);
|
|
|
|
// Let's calculate the correlation.
|
|
var correlation = math.correlation(tweets, followers);
|
|
approx.equal(correlation, 0.98054753183666);
|
|
|
|
// Now let's create a linear regression.
|
|
var linReg = math.linearRegression(tweets, followers);
|
|
|
|
// linReg is actually a function we can use to map tweets
|
|
// onto followers. We'll see that around 1422 followers
|
|
// are expected if a user tweets 500 times.
|
|
var estFollowers = linReg(500);
|
|
approx.equal(estFollowers, 1422.431464053916);
|
|
});
|
|
|
|
}); |