mathjs/test/expression/node/ArrayNode.test.js
2014-03-01 15:55:20 +01:00

99 lines
3.4 KiB
JavaScript

// test ArrayNode
var assert = require('assert'),
approx = require('../../../tools/approx'),
mathjs = require('../../../index'),
math = mathjs(),
Node = require('../../../lib/expression/node/Node'),
ConstantNode = require('../../../lib/expression/node/ConstantNode'),
SymbolNode = require('../../../lib/expression/node/SymbolNode'),
RangeNode = require('../../../lib/expression/node/RangeNode'),
ArrayNode = require('../../../lib/expression/node/ArrayNode');
describe('ArrayNode', function() {
it ('should create an ArrayNode', function () {
var c = new ConstantNode('number', '1');
var a = new ArrayNode([c]);
var b = new ArrayNode([]);
assert(a instanceof ArrayNode);
assert(b instanceof ArrayNode);
});
it ('should throw an error when calling without new operator', function () {
assert.throws(function () {ArrayNode()}, SyntaxError);
});
it ('should throw an error on wrong constructor arguments', function () {
assert.throws(function () {new ArrayNode(2)}, TypeError);
assert.throws(function () {new ArrayNode([2, 3])}, TypeError);
});
it ('should evaluate an ArrayNode', function () {
var c = new ConstantNode('number', '1');
var a = new ArrayNode([c]);
var b = new ArrayNode();
assert.deepEqual(a.compile(math).eval(), math.matrix([1]));
assert.deepEqual(b.compile(math).eval(), math.matrix([]));
});
it ('should compile an ArrayNode', function () {
var a = new ConstantNode('number', '1');
var b = new ConstantNode('number', '2');
var c = new ConstantNode('number', '3');
var d = new ConstantNode('number', '4');
var n = new ArrayNode([a, b, c, d]);
var expr = n.compile(math);
assert.deepEqual(expr.eval(), math.matrix([1,2,3,4]));
var mathArray = mathjs({matrix: 'array'});
var expr2 = n.compile(mathArray);
assert.deepEqual(expr2.eval(), [1,2,3,4]);
});
it ('should compile nested ArrayNodes', function () {
var a = new ConstantNode('number', '1');
var b = new ConstantNode('number', '2');
var c = new ConstantNode('number', '3');
var d = new ConstantNode('number', '4');
var n2 = new ArrayNode([a, b]);
var n3 = new ArrayNode([c, d]);
var n4 = new ArrayNode([n2, n3]);
var expr = n4.compile(math);
assert.deepEqual(expr.eval(), math.matrix([[1,2],[3,4]]));
});
it ('should find an ArrayNode', function () {
var a = new ConstantNode('number', '1');
var b = new SymbolNode('x');
var c = new ConstantNode('number', '2');
var d = new ArrayNode([a, b, c]);
assert.deepEqual(d.find({type: ArrayNode}), [d]);
assert.deepEqual(d.find({type: SymbolNode}), [b]);
assert.deepEqual(d.find({type: RangeNode}), []);
assert.deepEqual(d.find({type: ConstantNode}), [a, c]);
assert.deepEqual(d.find({type: ConstantNode, properties: {value: '2'}}), [c]);
});
it ('should match an ArrayNode', function () {
var a = new ArrayNode();
assert.equal(a.match({type: ArrayNode}), true);
assert.equal(a.match({type: SymbolNode}), false);
});
it ('should stringify an ArrayNode', function () {
var a = new ConstantNode('number', '1');
var b = new ConstantNode('number', '2');
var c = new ConstantNode('number', '3');
var d = new ConstantNode('number', '4');
var n = new ArrayNode([a, b, c, d]);
assert.equal(n.toString(), '[1, 2, 3, 4]');
});
});