archive-gh-me/mathjs
1
0
Fork 0
mirror of https://github.com/josdejong/mathjs.git synced 2025-12-08 19:46:04 +00:00
Code Issues Projects Releases Wiki Activity
mathjs/test/expression/node/OperatorNode.test.js
Max Bruckner 8d3b3bbe35 Tests for custom toString
2015-05-16 13:29:20 +02:00

529 lines
19 KiB
JavaScript
Raw Blame History

// test OperatorNode
var assert = require('assert');
var approx = require('../../../tools/approx');
var math = require('../../../index');
var Node = math.expression.node.Node;
var ConstantNode = math.expression.node.ConstantNode;
var SymbolNode = math.expression.node.SymbolNode;
var OperatorNode = math.expression.node.OperatorNode;
var ConditionalNode = math.expression.node.ConditionalNode;
describe('OperatorNode', function() {
it ('should create an OperatorNode', function () {
var n = new OperatorNode('op', 'fn', []);
assert(n instanceof OperatorNode);
assert(n instanceof Node);
assert.equal(n.type, 'OperatorNode');
});
it ('should have isOperatorNode', function () {
var node = new OperatorNode('op', 'fn', []);
assert(node.isOperatorNode);
});
it ('should throw an error when calling without new operator', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
assert.throws(function () {OperatorNode('+', 'add', [a, b])}, SyntaxError);
});
it ('should compile an OperatorNode', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var n = new OperatorNode('+', 'add', [a, b]);
var expr = n.compile(math);
assert.equal(expr.eval(), 5);
});
it ('should throw an error in case of unresolved operator function', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var n = new OperatorNode('+', 'add', [a, b]);
var emptyNamespace = {};
assert.throws(function () {
n.compile(emptyNamespace);
}, /Function add missing in provided namespace/);
});
it ('should filter an OperatorNode', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var n = new OperatorNode('+', 'add', [a, b]);
assert.deepEqual(n.filter(function (node) {return node instanceof OperatorNode}), [n]);
assert.deepEqual(n.filter(function (node) {return node instanceof SymbolNode}), []);
assert.deepEqual(n.filter(function (node) {return node instanceof ConstantNode}), [a, b]);
assert.deepEqual(n.filter(function (node) {return node instanceof ConstantNode && node.value == '2'}), [a]);
assert.deepEqual(n.filter(function (node) {return node instanceof ConstantNode && node.value == '4'}), []);
});
it ('should filter an OperatorNode without contents', function () {
var n = new OperatorNode('op', 'fn', []);
assert.deepEqual(n.filter(function (node) {return node instanceof OperatorNode}), [n]);
assert.deepEqual(n.filter(function (node) {return node instanceof SymbolNode}), []);
});
it ('should run forEach on an OperatorNode', function () {
// x^2-x
var a = new SymbolNode('x');
var b = new ConstantNode(2);
var c = new OperatorNode('^', 'pow', [a, b]);
var d = new SymbolNode('x');
var e = new OperatorNode('-', 'subtract', [c, d]);
var nodes = [];
var paths = [];
e.forEach(function (node, path, parent) {
nodes.push(node);
paths.push(path);
assert.strictEqual(parent, e);
});
assert.equal(nodes.length, 2);
assert.strictEqual(nodes[0], c);
assert.strictEqual(nodes[1], d);
assert.deepEqual(paths, ['args[0]', 'args[1]']);
});
it ('should map an OperatorNode', function () {
// x^2-x
var a = new SymbolNode('x');
var b = new ConstantNode(2);
var c = new OperatorNode('^', 'pow', [a, b]);
var d = new SymbolNode('x');
var e = new OperatorNode('-', 'subtract', [c, d]);
var nodes = [];
var paths = [];
var f = new ConstantNode(3);
var g = e.map(function (node, path, parent) {
nodes.push(node);
paths.push(path);
assert.strictEqual(parent, e);
return node instanceof SymbolNode && node.name == 'x' ? f : node;
});
assert.equal(nodes.length, 2);
assert.strictEqual(nodes[0], c);
assert.strictEqual(nodes[1], d);
assert.deepEqual(paths, ['args[0]', 'args[1]']);
assert.notStrictEqual(g, e);
assert.strictEqual(g.args[0], e.args[0]);
assert.strictEqual(g.args[0].args[0], a); // nested x is not replaced
assert.deepEqual(g.args[0].args[1], b);
assert.deepEqual(g.args[1], f);
});
it ('should throw an error when the map callback does not return a node', function () {
var a = new SymbolNode('x');
var b = new ConstantNode(2);
var c = new OperatorNode('^', 'pow', [a, b]);
assert.throws(function () {
c.map(function () {});
}, /Callback function must return a Node/)
});
it ('should transform an OperatorNodes parameters', function () {
// x^2-x
var a = new SymbolNode('x');
var b = new ConstantNode(2);
var c = new OperatorNode('^', 'pow', [a, b]);
var d = new SymbolNode('x');
var e = new OperatorNode('-', 'subtract', [c, d]);
var f = new ConstantNode(3);
var g = e.transform(function (node) {
return node instanceof SymbolNode && node.name == 'x' ? f : node;
});
assert.notStrictEqual(g, e);
assert.notStrictEqual(g.args[0], e.args[0]);
assert.strictEqual(g.args[0].args[0], f);
assert.deepEqual(g.args[0].args[1], b);
assert.deepEqual(g.args[1], f);
});
it ('should transform an OperatorNode itself', function () {
// x^2-x
var a = new SymbolNode('x');
var b = new ConstantNode(2);
var c = new OperatorNode('+', 'add', [a, b]);
var f = new ConstantNode(3);
var g = c.transform(function (node) {
return node instanceof OperatorNode ? f : node;
});
assert.notStrictEqual(g, c);
assert.deepEqual(g, f);
});
it ('should clone an OperatorNode', function () {
// x^2-x
var a = new SymbolNode('x');
var b = new ConstantNode(2);
var c = new OperatorNode('+', 'add', [a, b]);
var d = c.clone();
assert(d instanceof OperatorNode);
assert.deepEqual(d, c);
assert.notStrictEqual(d, c);
assert.notStrictEqual(d.args, c.args);
assert.strictEqual(d.args[0], c.args[0]);
assert.strictEqual(d.args[1], c.args[1]);
});
describe('toString', function () {
it ('should stringify an OperatorNode', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var c = new ConstantNode(4);
var n = new OperatorNode('+', 'add', [a, b]);
assert.equal(n.toString(), '2 + 3');
});
it ('should stringify an OperatorNode with factorial', function () {
var a = new ConstantNode(2);
var n = new OperatorNode('!', 'factorial', [a]);
assert.equal(n.toString(), '2!');
});
it ('should stringify an OperatorNode with unary minus', function () {
var a = new ConstantNode(2);
var n = new OperatorNode('-', 'unaryMinus', [a]);
assert.equal(n.toString(), '-2');
});
it ('should stringify an OperatorNode with zero arguments', function () {
var n = new OperatorNode('foo', 'foo', []);
assert.equal(n.toString(), 'foo()');
});
it ('should stringify an OperatorNode with more than two operators', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var c = new ConstantNode(4);
var n = new OperatorNode('foo', 'foo', [a, b, c]);
assert.equal(n.toString(), 'foo(2, 3, 4)');
});
it ('should stringify an OperatorNode with nested operator nodes', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var c = new ConstantNode(4);
var d = new ConstantNode(5);
var n1 = new OperatorNode('+', 'add', [a, b]);
var n2 = new OperatorNode('-', 'subtract', [c, d]);
var n3 = new OperatorNode('*', 'multiply', [n1, n2]);
assert.equal(n1.toString(), '2 + 3');
assert.equal(n2.toString(), '4 - 5');
assert.equal(n3.toString(), '(2 + 3) * (4 - 5)');
});
it ('should stringify left associative OperatorNodes that are associative with another Node', function () {
var autoMath = math.create({parenthesis: 'auto'});
assert.equal(autoMath.parse('(a+b)+c').toString(), 'a + b + c');
assert.equal(autoMath.parse('a+(b+c)').toString(), 'a + b + c');
assert.equal(autoMath.parse('(a+b)-c').toString(), 'a + b - c');
assert.equal(autoMath.parse('a+(b-c)').toString(), 'a + b - c');
assert.equal(autoMath.parse('(a*b)*c').toString(), 'a * b * c');
assert.equal(autoMath.parse('a*(b*c)').toString(), 'a * b * c');
assert.equal(autoMath.parse('(a*b)/c').toString(), 'a * b / c');
assert.equal(autoMath.parse('a*(b/c)').toString(), 'a * b / c');
});
it ('should stringify left associative OperatorNodes that are not associative with another Node', function () {
var autoMath = math.create({parenthesis: 'auto'});
assert.equal(autoMath.parse('(a-b)-c').toString(), 'a - b - c');
assert.equal(autoMath.parse('a-(b-c)').toString(), 'a - (b - c)');
assert.equal(autoMath.parse('(a-b)+c').toString(), 'a - b + c');
assert.equal(autoMath.parse('a-(b+c)').toString(), 'a - (b + c)');
assert.equal(autoMath.parse('(a/b)/c').toString(), 'a / b / c');
assert.equal(autoMath.parse('a/(b/c)').toString(), 'a / (b / c)');
assert.equal(autoMath.parse('(a/b)*c').toString(), 'a / b * c');
assert.equal(autoMath.parse('a/(b*c)').toString(), 'a / (b * c)');
});
it ('should stringify right associative OperatorNodes that are not associative with another Node', function () {
var autoMath = math.create({parenthesis: 'auto'});
assert.equal(autoMath.parse('(a^b)^c').toString(), '(a ^ b) ^ c');
assert.equal(autoMath.parse('a^(b^c)').toString(), 'a ^ b ^ c');
});
it ('should stringify unary OperatorNodes containing a binary OperatorNode', function () {
assert.equal(math.parse('(a*b)!').toString(), '(a * b)!');
assert.equal(math.parse('-(a*b)').toString(), '-(a * b)');
assert.equal(math.parse('-(a+b)').toString(), '-(a + b)');
});
it ('should stringify unary OperatorNodes containing a unary OperatorNode', function () {
var autoMath = math.create({parenthesis: 'auto'});
assert.equal(autoMath.parse('(-a)!').toString(), '(-a)!');
assert.equal(autoMath.parse('-(a!)').toString(), '-a!');
assert.equal(autoMath.parse('-(-a)').toString(), '-(-a)');
});
});
it ('should stringify an OperatorNode with custom toString', function () {
//Also checks if the custom functions get passed on to the children
var customFunction = function (node, config, callback) {
if (node.type === 'OperatorNode') {
return node.op + node.fn + '('
+ node.args[0].toString(config, callback)
+ ', ' + node.args[1].toString(config, callback) + ')';
}
else if (node.type === 'ConstantNode') {
return 'const(' + node.value + ', ' + node.valueType + ')'
}
};
var a = new ConstantNode(1);
var b = new ConstantNode(2);
var n1 = new OperatorNode('+', 'add', [a, b]);
var n2 = new OperatorNode('-', 'subtract', [a, b]);
assert.equal(n1.toString({}, customFunction), '+add(const(1, number), const(2, number))');
assert.equal(n2.toString({}, customFunction), '-subtract(const(1, number), const(2, number))');
});
it ('should stringify an OperatorNode with custom toString for a single operator', function () {
//Also checks if the custom functions get passed on to the children
var customFunction = function (node, config, callback) {
if ((node.type === 'OperatorNode') && (node.fn === 'add')) {
return node.args[0].toString(config, callback)
+ node.op + node.fn + node.op +
node.args[1].toString(config, callback);
}
else if (node.type === 'ConstantNode') {
return 'const(' + node.value + ', ' + node.valueType + ')'
}
};
var a = new ConstantNode(1);
var b = new ConstantNode(2);
var n = new OperatorNode('+', 'add', [a, b]);
assert.equal(n.toString({}, customFunction), 'const(1, number)+add+const(2, number)');
});
it ('should respect the \'all\' parenthesis option', function () {
var allMath = math.create({parenthesis: 'all'});
assert.equal(allMath.parse('1+1+1').toString(), '(1 + 1) + 1' );
assert.equal(allMath.parse('1+1+1').toTex(), '\\left(1+1\\right)+1' );
});
it ('should correctly LaTeX fractions in \'all\' parenthesis mode', function () {
var allMath = math.create({parenthesis: 'all'});
assert.equal(allMath.parse('1/2/3').toTex(), '\\frac{\\left(\\frac{1}{2}\\right)}{3}');
});
it ('should LaTeX an OperatorNode', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var c = new ConstantNode(4);
var n = new OperatorNode('+', 'add', [a, b]);
assert.equal(n.toTex(), '2+3');
});
it ('should LaTeX an OperatorNode with factorial', function () {
var a = new ConstantNode(2);
var n = new OperatorNode('!', 'factorial', [a]);
assert.equal(n.toTex(), '2!');
});
it ('should LaTeX an OperatorNode with factorial of an OperatorNode', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var sub = new OperatorNode('-', 'subtract', [a, b]);
var add = new OperatorNode('+', 'add', [a, b]);
var mult = new OperatorNode('*', 'multiply', [a, b]);
var div = new OperatorNode('/', 'divide', [a, b]);
var n1= new OperatorNode('!', 'factorial', [sub] );
var n2= new OperatorNode('!', 'factorial', [add] );
var n3= new OperatorNode('!', 'factorial', [mult] );
var n4= new OperatorNode('!', 'factorial', [div] );
assert.equal(n1.toTex(), '\\left(2-3\\right)!');
assert.equal(n2.toTex(), '\\left(2+3\\right)!');
assert.equal(n3.toTex(), '\\left(2\\cdot3\\right)!');
assert.equal(n4.toTex(), '\\frac{2}{3}!');
});
it ('should LaTeX an OperatorNode with unary minus', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var sub = new OperatorNode('-', 'subtract', [a, b]);
var add = new OperatorNode('+', 'add', [a, b]);
var n1 = new OperatorNode('-', 'unaryMinus', [a]);
var n2 = new OperatorNode('-', 'unaryMinus', [sub]);
var n3 = new OperatorNode('-', 'unaryMinus', [add]);
assert.equal(n1.toTex(), '-2');
assert.equal(n2.toTex(), '-\\left(2-3\\right)');
assert.equal(n3.toTex(), '-\\left(2+3\\right)');
});
it ('should LaTeX an OperatorNode that subtracts an OperatorNode', function() {
var a = new ConstantNode(1);
var b = new ConstantNode(2);
var c = new ConstantNode(3);
var sub = new OperatorNode('-', 'subtract', [b, c]);
var add = new OperatorNode('+', 'add', [b, c]);
var n1 = new OperatorNode('-', 'subtract', [a, sub]);
var n2 = new OperatorNode('-', 'subtract', [a, add]);
assert.equal(n1.toTex(), '1-\\left(2-3\\right)');
assert.equal(n2.toTex(), '1-\\left(2+3\\right)');
});
it ('should LaTeX an OperatorNode with zero arguments', function () {
var n = new OperatorNode('foo', 'foo', []);
assert.equal(n.toTex(), '\\mathrm{foo}\\left(\\right)');
});
it ('should LaTeX an OperatorNode with more than two operators', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var c = new ConstantNode(4);
var n = new OperatorNode('foo', 'foo', [a, b, c]);
assert.equal(n.toTex(), '\\mathrm{foo}\\left(2,3,4\\right)');
});
it ('should LaTeX an OperatorNode with nested operator nodes', function () {
var a = new ConstantNode(2);
var b = new ConstantNode(3);
var c = new ConstantNode(4);
var d = new ConstantNode(5);
var n1 = new OperatorNode('+', 'add', [a, b]);
var n2 = new OperatorNode('-', 'subtract', [c, d]);
var n3 = new OperatorNode('*', 'multiply', [n1, n2]);
var m2 = new OperatorNode('*', 'multiply', [n1, c]);
var m3 = new OperatorNode('-', 'subtract', [m2, d]);
assert.equal(n1.toTex(), '2+3');
assert.equal(n2.toTex(), '4-5');
assert.equal(n3.toTex(), '\\left(2+3\\right)\\cdot\\left(4-5\\right)');
assert.equal(m3.toTex(), '\\left(2+3\\right)\\cdot4-5');
});
it('should LaTeX fractions with operators that are enclosed in parenthesis', function () {
var a = new ConstantNode(1);
var b = new ConstantNode(2);
var add = new OperatorNode('+', 'add', [a,a]);
var frac = new OperatorNode('/', 'divide', [add,b]);
assert.equal(frac.toTex(), '\\frac{1+1}{2}');
});
it ('should have an identifier', function () {
var a = new ConstantNode(1);
var b = new ConstantNode(2);
var n = new OperatorNode('+', 'add', [a, b]);
assert.equal(n.getIdentifier(), 'OperatorNode:add');
});
it ('should LaTeX an OperatorNode with custom toTex', function () {
//Also checks if the custom functions get passed on to the children
var customFunction = function (node, config, callback) {
if (node.type === 'OperatorNode') {
return node.op + node.fn + '('
+ node.args[0].toTex(config, callback)
+ ', ' + node.args[1].toTex(config, callback) + ')';
}
else if (node.type === 'ConstantNode') {
return 'const\\left(' + node.value + ', ' + node.valueType + '\\right)'
}
};
var a = new ConstantNode(1);
var b = new ConstantNode(2);
var n1 = new OperatorNode('+', 'add', [a, b]);
var n2 = new OperatorNode('-', 'subtract', [a, b]);
assert.equal(n1.toTex({}, customFunction), '+add(const\\left(1, number\\right), const\\left(2, number\\right))');
assert.equal(n2.toTex({}, customFunction), '-subtract(const\\left(1, number\\right), const\\left(2, number\\right))');
});
it ('should LaTeX an OperatorNode with custom toTex for a single operator', function () {
//Also checks if the custom functions get passed on to the children
var customFunction = function (node, config, callback) {
if ((node.type === 'OperatorNode') && (node.fn === 'add')) {
return node.args[0].toTex(config, callback)
+ node.op + node.fn + node.op +
node.args[1].toTex(config, callback);
}
else if (node.type === 'ConstantNode') {
return 'const\\left(' + node.value + ', ' + node.valueType + '\\right)'
}
};
var a = new ConstantNode(1);
var b = new ConstantNode(2);
var n = new OperatorNode('+', 'add', [a, b]);
assert.equal(n.toTex({}, customFunction), 'const\\left(1, number\\right)+add+const\\left(2, number\\right)');
});
it ('should LaTeX powers of fractions with parentheses', function () {
var a = new ConstantNode(1);
var frac = new OperatorNode('/', 'divide', [a,a]);
var pow = new OperatorNode('^', 'pow', [frac, a]);
assert.equal(pow.toTex(), '\\left({\\frac{1}{1}}\\right)^{1}');
});
it ('should LaTeX powers of conditions with parentheses', function () {
var a = new ConstantNode(1);
var cond = new ConditionalNode(a, a, a);
var pow = new OperatorNode('^', 'pow', [cond, a]);
assert.equal(pow.toTex(), '\\left({\\left\\{\\begin{array}{l l}{1}, &\\quad{\\text{if}\\;1}\\\\{1}, &\\quad{\\text{otherwise}}\\end{array}\\right.}\\right)^{1}');
});
it ('should LaTeX simple expressions in \'auto\' mode', function () {
var autoMath = math.create({parenthesis: 'auto'});
//this covers a bug that was triggered previously
assert.equal(autoMath.parse('1+(1+1)').toTex(), '1+1+1');
});
});
Reference in New Issue View Git Blame Copy Permalink