mirror of
https://github.com/josdejong/mathjs.git
synced 2026-01-18 14:59:29 +00:00
80 lines
2.4 KiB
JavaScript
80 lines
2.4 KiB
JavaScript
// test ones
|
|
var assert = require('assert'),
|
|
error = require('../../../lib/error/index'),
|
|
math = require('../../../index'),
|
|
ones = math.ones,
|
|
matrix = math.matrix;
|
|
|
|
describe('ones', function() {
|
|
|
|
it('should create an empty matrix', function () {
|
|
assert.deepEqual(ones(), matrix());
|
|
assert.deepEqual(ones([]), []);
|
|
assert.deepEqual(ones(matrix([])), matrix());
|
|
});
|
|
|
|
it('should create an empty matrix, CCS format', function () {
|
|
assert.deepEqual(ones('ccs'), matrix('ccs'));
|
|
assert.deepEqual(ones([], 'ccs'), matrix([], 'ccs'));
|
|
assert.deepEqual(ones(matrix([]), 'ccs'), matrix('ccs'));
|
|
});
|
|
|
|
it('should create an empty matrix, CRS format', function () {
|
|
assert.deepEqual(ones('crs'), matrix('crs'));
|
|
assert.deepEqual(ones([], 'crs'), matrix([], 'crs'));
|
|
assert.deepEqual(ones(matrix([]), 'crs'), matrix('crs'));
|
|
});
|
|
|
|
it('should create a vector with ones', function () {
|
|
assert.deepEqual(ones(3), matrix([1,1,1]));
|
|
assert.deepEqual(ones(matrix([4])), matrix([1,1,1,1]));
|
|
assert.deepEqual(ones([4]), [1,1,1,1]);
|
|
assert.deepEqual(ones(0), matrix([]));
|
|
});
|
|
|
|
it('should create a 2D matrix with ones', function () {
|
|
assert.deepEqual(ones(2,3), matrix([[1,1,1],[1,1,1]]));
|
|
assert.deepEqual(ones(3,2), matrix([[1,1],[1,1],[1,1]]));
|
|
assert.deepEqual(ones([3,2]), [[1,1],[1,1],[1,1]]);
|
|
});
|
|
|
|
it('should create a matrix with ones from a matrix', function () {
|
|
assert.deepEqual(ones(matrix([3])), matrix([1,1,1]));
|
|
assert.deepEqual(ones(matrix([3,2])), matrix([[1,1],[1,1],[1,1]]));
|
|
});
|
|
|
|
it('should create a matrix with bignumber ones', function () {
|
|
var one = math.bignumber(1);
|
|
var three = math.bignumber(3);
|
|
assert.deepEqual(ones(three), matrix([one,one,one]));
|
|
assert.deepEqual(ones([three]), [one,one,one]);
|
|
});
|
|
|
|
it('should create a 3D matrix with ones', function () {
|
|
var res = [
|
|
[
|
|
[1,1,1,1],
|
|
[1,1,1,1],
|
|
[1,1,1,1]
|
|
],
|
|
[
|
|
[1,1,1,1],
|
|
[1,1,1,1],
|
|
[1,1,1,1]
|
|
]
|
|
];
|
|
assert.deepEqual(ones(2,3,4), matrix(res));
|
|
assert.deepEqual(ones(matrix([2,3,4])), matrix(res));
|
|
assert.deepEqual(ones([2,3,4]), res);
|
|
});
|
|
|
|
// TODO: test setting `matrix`
|
|
|
|
it('should create a matrix with ones with the same size as original matrix', function () {
|
|
var a = matrix([[1, 2, 3], [4, 5, 6]]);
|
|
assert.deepEqual(ones(math.size(a)).size(), a.size());
|
|
});
|
|
|
|
// TODO: test with invalid input
|
|
|
|
}); |