mathjs/test/function/matrix/subset.test.js

93 lines
3.9 KiB
JavaScript

var assert = require('assert'),
math = require('../../../index.js'),
subset = math.subset,
matrix = math.matrix,
range = math.range,
index = math.index;
describe('subset', function() {
var a = [[1,2], [3,4]];
var b = math.matrix(a);
it('should get the right subset of an array', function() {
assert.deepEqual(subset(a, index([0,2], 1)), [[2],[4]]);
assert.deepEqual(subset(a, index(1,0)), 3);
});
it('should throw an error if trying to access an invalid subset of an array', function() {
assert.throws(function () {subset(a, index(6, 0))}, RangeError);
assert.throws(function () {subset(a, index(1))}, RangeError);
assert.throws(function () {subset(a, index(1,0,0))}, RangeError);
assert.throws(function () {subset(a, index(1.3, 0))}, TypeError);
});
it('should get the right subset of a matrix', function() {
assert.deepEqual(subset(b, index([0,2], 1)), matrix([[2],[4]]));
assert.deepEqual(subset(b, index(1, 0)), 3);
});
it('should throw an error if trying to access an invalid subset of a matrix', function() {
assert.throws(function () {subset(b, index(6, 0))}, RangeError);
assert.throws(function () {subset(b, index(1))}, RangeError);
assert.throws(function () {subset(b, index(1,0,0))}, RangeError);
assert.throws(function () {subset(b, index(1.3, 0))}, TypeError);
});
it('should get the right subset of a string', function() {
assert.deepEqual(subset('hello', index(1)), 'e');
assert.deepEqual(subset('hello', index([4,-1,-1])), 'olleh');
});
it('should throw an error if trying to access an invalid subset of a string', function() {
assert.throws(function () {subset('hello', 1)}, TypeError);
assert.throws(function () {subset('hello', index([6]))}, SyntaxError);
assert.throws(function () {subset('hello', index([-2]))}, SyntaxError);
assert.throws(function () {subset('hello', index([1.3]))}, TypeError);
});
var d = [[1,2], [3,4]];
var g = matrix([[1,2], [3,4]]);
it('should set the right subset of an array', function() {
assert.deepEqual(d, [[1,2], [3,4]]);
assert.deepEqual(subset(d, index([0,2], 1), [[-2],[-4]]), [[1,-2], [3,-4]]);
assert.deepEqual(d, [[1,2], [3,4]]);
assert.deepEqual(subset(d, index(2, [0,2]), [[5,6]]), [[1,2], [3,4], [5,6]]);
assert.deepEqual(d, [[1,2], [3,4]]);
assert.deepEqual(subset(d, index(0,0), 123), [[123,2], [3,4]]);
});
it('should throw an error if setting the subset of an array with an invalid replacement', function() {
assert.throws(function () {subset(d, index(1), 123)}, RangeError);
assert.throws(function () {subset(d, index(1.3,0), 123)}, TypeError);
});
it('should set the right subset of a matrix', function() {
assert.deepEqual(g, matrix([[1,2], [3,4]]));
assert.deepEqual(subset(g, index([0,2], 1), [[-2],[-4]]), matrix([[1,-2], [3,-4]]));
assert.deepEqual(g, matrix([[1,2], [3,4]]));
assert.deepEqual(subset(g, index(2, [0,2]), [[5,6]]), matrix([[1,2], [3,4], [5,6]]));
});
it('should throw an error if setting the subset of a matrix with an invalid replacement', function() {
assert.throws(function () {subset(d, index(1), 123)}, RangeError);
assert.throws(function () {subset(d, index(1.3,0), 123)}, TypeError);
});
it('should set the right subset of a string', function() {
var j = 'hello';
assert.deepEqual(subset(j, index(0), 'H'), 'Hello');
assert.deepEqual(j, 'hello');
assert.deepEqual(subset(j, index(5), '!'), 'hello!');
assert.deepEqual(j, 'hello');
assert.deepEqual(subset(j, index([5,11]), ' world'), 'hello world');
assert.deepEqual(j, 'hello');
});
it('should throw an error if setting the subset of a string with an invalid replacement', function() {
assert.throws(function () {subset('hello', index([1,2]), '1234')}, RangeError);
assert.throws(function () {subset('hello', index(1,2), 'a')}, RangeError);
});
});