mirror of
https://github.com/josdejong/mathjs.git
synced 2026-01-18 14:59:29 +00:00
15 lines
802 B
JavaScript
15 lines
802 B
JavaScript
var assert = require('assert');
|
|
var math = require('../../../src/index.js');
|
|
|
|
// test diag
|
|
assert.deepEqual(math.diag([1,2,3]).valueOf(), [[1,0,0],[0,2,0],[0,0,3]]);
|
|
assert.deepEqual(math.diag([1,2,3], 1).valueOf(), [[0,1,0,0],[0,0,2,0],[0,0,0,3]]);
|
|
assert.deepEqual(math.diag([1,2,3], -1).valueOf(), [[0,0,0],[1,0,0],[0,2,0],[0,0,3]]);
|
|
assert.deepEqual(math.diag([[1,2,3],[4,5,6]]).valueOf(), [1,5]);
|
|
assert.deepEqual(math.diag([[1,2,3],[4,5,6]],1).valueOf(), [2,6]);
|
|
assert.deepEqual(math.diag([[1,2,3],[4,5,6]],-1).valueOf(), [4]);
|
|
assert.deepEqual(math.diag([[1,2,3],[4,5,6]],-2).valueOf(), []);
|
|
assert.deepEqual(math.diag(math.range(1,3)).valueOf(), [[1,0,0],[0,2,0],[0,0,3]]);
|
|
assert.throws(function () {math.diag([[[1],[2]],[[3],[4]]])});
|
|
// TODO: test diag for all types of input (also scalar)
|