mirror of
https://github.com/josdejong/mathjs.git
synced 2025-12-08 19:46:04 +00:00
114 lines
3.2 KiB
JavaScript
114 lines
3.2 KiB
JavaScript
var assert = require('assert'),
|
|
error = require('../../../lib/error/index'),
|
|
approx = require('../../../tools/approx'),
|
|
math = require('../../../index');
|
|
|
|
describe('trace', function() {
|
|
|
|
it('should calculate correctly the trace of a NxN matrix', function() {
|
|
assert.equal(math.trace([5]), 5);
|
|
assert.equal(math.trace([[1,2],[3,4]]), 5);
|
|
assert.equal(math.trace(math.matrix([[1,2],[3,4]])), 5);
|
|
approx.equal(math.trace([
|
|
[-2, 2, 3],
|
|
[-1, 1, 3],
|
|
[ 2, 0, -1]
|
|
]), -2);
|
|
approx.equal(math.trace([
|
|
[ 1, 4, 7],
|
|
[ 3, 0, 5],
|
|
[-1, 9, 11]
|
|
]), 12);
|
|
approx.equal(math.trace([
|
|
[1,7,4,3,7],
|
|
[0,7,0,3,7],
|
|
[0,7,4,3,0],
|
|
[1,7,5,9,7],
|
|
[2,7,4,3,7]
|
|
]), 28);
|
|
approx.equal(math.trace(math.diag([4,-5,6])), 5);
|
|
});
|
|
|
|
it('should return N for the identity matrix',function() {
|
|
assert.equal(math.trace(math.eye(7)), 7);
|
|
assert.equal(math.trace(math.eye(2)), 2);
|
|
assert.equal(math.trace(math.eye(1)), 1);
|
|
});
|
|
|
|
it('should calculate the trace for a scalar',function() {
|
|
assert.equal(math.trace(7), 7);
|
|
|
|
var c1 = math.complex(2, 3);
|
|
var c2 = math.trace(c1);
|
|
assert.deepEqual(c1, c2);
|
|
|
|
// c2 should be a clone
|
|
c1.re = 0;
|
|
assert.equal(c1.re, 0);
|
|
assert.equal(c2.re, 2);
|
|
});
|
|
|
|
it('should calculate the trace for a 1x1 matrix',function() {
|
|
var c1 = math.complex(2, 3);
|
|
var c2 = math.trace([[c1]]);
|
|
assert.deepEqual(c1, c2);
|
|
|
|
// c2 should be a clone
|
|
c1.re = 0;
|
|
assert.equal(c1.re, 0);
|
|
assert.equal(c2.re, 2);
|
|
});
|
|
|
|
it('should calculate correctly the trace of a matrix with bignumbers', function() {
|
|
var bignumber = math.bignumber;
|
|
|
|
// 1x1
|
|
assert.deepEqual(math.trace([bignumber(5)]), bignumber(5));
|
|
|
|
// 2x2
|
|
assert.deepEqual(math.trace([
|
|
[bignumber(1), bignumber(2)],
|
|
[bignumber(3), bignumber(4)]
|
|
]), bignumber(5));
|
|
|
|
// 3x3
|
|
assert.deepEqual(math.trace([
|
|
[bignumber(-2), bignumber(2), bignumber( 3)],
|
|
[bignumber(-1), bignumber(1), bignumber( 3)],
|
|
[bignumber( 2), bignumber(0), bignumber(-1)]
|
|
]), bignumber(-2));
|
|
|
|
// the following would fail with regular Numbers due to a precision overflow
|
|
assert.deepEqual(math.trace([
|
|
[bignumber(1e10+1), bignumber(1e10)],
|
|
[bignumber(1e10), bignumber(-1e10)]
|
|
]), bignumber(1));
|
|
});
|
|
|
|
it('should calculate the trace of a matrix with mixed numbers and bignumbers', function() {
|
|
var bignumber = math.bignumber;
|
|
assert.deepEqual(math.trace([
|
|
[bignumber(2), 1],
|
|
[bignumber(3), 4]
|
|
]), bignumber(6));
|
|
});
|
|
|
|
it('should not change the value of the initial matrix', function() {
|
|
var m_test = [[1,2,3],[4,5,6],[7,8,9]];
|
|
math.trace(m_test);
|
|
assert.deepEqual(m_test, [[1,2,3],[4,5,6],[7,8,9]]);
|
|
});
|
|
|
|
it('should not accept a non-square matrix', function() {
|
|
assert.throws(function() { math.trace([1,2]); });
|
|
assert.throws(function() { math.trace([[1,2,3],[1,2,3]]); });
|
|
assert.throws(function() { math.trace([0,1],[0,1],[0,1]); });
|
|
});
|
|
|
|
it('should not accept arrays with dimensions higher than 2', function() {
|
|
assert.throws(function() { math.trace([[[1]]]); }, RangeError);
|
|
assert.throws(function() { math.trace(math.matrix([[[1]]])); }, RangeError);
|
|
});
|
|
|
|
|
|
}); |