632 lines
13 KiB
JavaScript
632 lines
13 KiB
JavaScript
var assert = require("assert");
|
|
|
|
var Complex = require("../complex");
|
|
|
|
var tests = [{
|
|
set: null,
|
|
expect: "0"
|
|
}, {
|
|
set: undefined,
|
|
expect: "0"
|
|
}, {
|
|
set: "foo",
|
|
expect: "SyntaxError: Invalid Param"
|
|
}, {
|
|
set: {},
|
|
expect: "SyntaxError: Invalid Param"
|
|
}, {
|
|
set: " + i",
|
|
expect: "i"
|
|
}, {
|
|
set: "3",
|
|
expect: "3"
|
|
}, {
|
|
set: "2.3",
|
|
expect: "2.3"
|
|
}, {
|
|
set: {re: -Infinity, im: 0},
|
|
expect: "-Infinity"
|
|
}, {
|
|
set: Complex.I,
|
|
fn: "mul",
|
|
param: Complex(Math.PI).exp(),
|
|
expect: "23.140692632779274i"
|
|
}, {
|
|
set: 0,
|
|
expect: "0"
|
|
}, {
|
|
set: "4 + 3i",
|
|
fn: "add",
|
|
param: "-3 - 2i",
|
|
expect: "1 + i"
|
|
}, {
|
|
set: "3i",
|
|
fn: "add",
|
|
param: "-2i",
|
|
expect: "i"
|
|
}, {
|
|
set: "4",
|
|
fn: "add",
|
|
param: "-3",
|
|
expect: "1"
|
|
}, {
|
|
set: 9,
|
|
fn: "sqrt",
|
|
expect: "3"
|
|
}, {
|
|
set: -9,
|
|
fn: "sqrt",
|
|
expect: "3i"
|
|
}, {
|
|
set: "-36",
|
|
fn: "sqrt",
|
|
expect: "6i"
|
|
}, {
|
|
set: "36i",
|
|
fn: "sqrt",
|
|
expect: "4.242640687119285 + 4.242640687119285i"
|
|
}, {
|
|
set: Infinity,
|
|
fn: "mul",
|
|
param: "i",
|
|
expect: "NaN"
|
|
}, {
|
|
set: "-36i",
|
|
fn: "sqrt",
|
|
expect: "4.242640687119285 - 4.242640687119285i"
|
|
}, {
|
|
set: "4 + 2i",
|
|
fn: "div",
|
|
param: "0",
|
|
expect: "Infinity + Infinityi"
|
|
}, {
|
|
set: "4 + 2i",
|
|
fn: "div",
|
|
param: "1 + i",
|
|
expect: "3 - i"
|
|
}, {
|
|
set: "25",
|
|
fn: "div",
|
|
param: "3 - 4i",
|
|
expect: "3 + 4i"
|
|
}, {
|
|
set: "3 - 2i",
|
|
fn: "div",
|
|
param: "i",
|
|
expect: "-2 - 3i"
|
|
}, {
|
|
set: "4i",
|
|
fn: "mul",
|
|
param: "-5i",
|
|
expect: "20"
|
|
}, {
|
|
set: "3 - 6i",
|
|
fn: "mul",
|
|
param: "i",
|
|
expect: "6 + 3i"
|
|
}, {
|
|
set: "3 + 4i",
|
|
fn: "add",
|
|
param: "5 - 7i",
|
|
expect: "8 - 3i"
|
|
}, {
|
|
set: "6i",
|
|
fn: "div",
|
|
param: "3 - 12i",
|
|
expect: "-0.47058823529411764 + 0.11764705882352941i"
|
|
}, {
|
|
set: "36 + 36i",
|
|
fn: "sqrt",
|
|
expect: "6.59210468080686 + 2.730539163373364i"
|
|
}, {
|
|
set: "36 - 36i",
|
|
fn: "sqrt",
|
|
expect: "6.59210468080686 - 2.730539163373364i"
|
|
}, {
|
|
set: "-36 + 36i",
|
|
fn: "sqrt",
|
|
expect: "2.730539163373364 + 6.59210468080686i"
|
|
}, {
|
|
set: "-36 - 36i",
|
|
fn: "sqrt",
|
|
expect: "2.730539163373364 - 6.59210468080686i"
|
|
}, {
|
|
set: "0",
|
|
fn: "sqrt",
|
|
expect: "0"
|
|
}, {
|
|
set: Math.E,
|
|
fn: "log",
|
|
expect: "1"
|
|
}, {
|
|
set: 0,
|
|
fn: "log",
|
|
expect: "-Infinity"
|
|
}, {
|
|
set: Infinity,
|
|
fn: "mul",
|
|
param: 3,
|
|
expect: "Infinity"
|
|
}, {
|
|
set: "-1",
|
|
fn: "log",
|
|
expect: Math.PI + "i"
|
|
}, {
|
|
set: "i",
|
|
fn: "log",
|
|
expect: (Math.PI / 2) + "i"
|
|
}, {
|
|
set: "3 + 2i",
|
|
fn: "log",
|
|
expect: Math.log(13) / 2 + " + " + Math.atan2(2, 3) + "i"
|
|
}, {
|
|
set: "3 - 2i",
|
|
fn: "log",
|
|
expect: Math.log(13) / 2 + " - " + Math.atan2(2, 3) + "i"
|
|
}, {
|
|
set: 1,
|
|
fn: "exp",
|
|
expect: "" + Math.E
|
|
}, {
|
|
set: "i",
|
|
fn: "exp",
|
|
expect: Math.cos(1) + " + " + Math.sin(1) + "i"
|
|
}, {
|
|
set: "3 + 2i",
|
|
fn: "exp",
|
|
expect: "-8.358532650935372 + 18.263727040666765i"
|
|
}, {
|
|
set: "3 - 2i",
|
|
fn: "exp",
|
|
expect: "-8.358532650935372 - 18.263727040666765i"
|
|
}, {
|
|
set: "3",
|
|
fn: "pow",
|
|
param: "3",
|
|
expect: "27"
|
|
}, {
|
|
set: "i",
|
|
fn: "pow",
|
|
param: "0",
|
|
expect: "1"
|
|
}, {
|
|
set: "87",
|
|
fn: "pow",
|
|
param: "3",
|
|
expect: "658503"
|
|
}, {
|
|
set: "i",
|
|
fn: "pow",
|
|
param: "1",
|
|
expect: "i"
|
|
}, {
|
|
set: "i",
|
|
fn: "pow",
|
|
param: "2",
|
|
expect: "-1"
|
|
}, {
|
|
set: "i",
|
|
fn: "pow",
|
|
param: "3",
|
|
expect: "-i"
|
|
}, {
|
|
set: "i",
|
|
fn: "pow",
|
|
param: "4",
|
|
expect: "1"
|
|
}, {
|
|
set: "i",
|
|
fn: "pow",
|
|
param: "5",
|
|
expect: "i"
|
|
}, {
|
|
set: 7,
|
|
fn: "pow",
|
|
param: 2,
|
|
expect: 49
|
|
}, {
|
|
set: 0,
|
|
fn: "pow",
|
|
param: 2,
|
|
expect: 0
|
|
}, {
|
|
set: "3i",
|
|
fn: "pow",
|
|
param: "3i",
|
|
expect: "-0.008876640735623675 - 0.0013801328997494896i"
|
|
}, {
|
|
set: {re: 3, im: 4},
|
|
fn: "abs",
|
|
expect: "5"
|
|
}, {
|
|
set: {re: 10, im: 24},
|
|
fn: "abs",
|
|
expect: "26"
|
|
}, {
|
|
set: "1 + 4i",
|
|
fn: "mul",
|
|
param: "3 + 2i",
|
|
expect: "-5 + 14i"
|
|
}, {
|
|
set: "4 + 16i",
|
|
fn: "div",
|
|
param: "4.0000",
|
|
expect: "1 + 4i"
|
|
}, {
|
|
set: {re: -7.1, im: 2.5},
|
|
fn: "neg",
|
|
expect: "7.1 - 2.5i"
|
|
}, {
|
|
set: {re: 1, im: 1},
|
|
fn: "arg",
|
|
expect: "" + Math.PI / 4
|
|
}, {
|
|
set: {re: -1, im: -1},
|
|
fn: "arg",
|
|
expect: "" + -3 * Math.PI / 4
|
|
}, {
|
|
set: {re: 0, im: 1},
|
|
fn: "arg",
|
|
expect: "" + Math.PI / 2
|
|
}, {
|
|
set: {re: 1, im: 0.5 * Math.sqrt(4 / 3)},
|
|
fn: "arg",
|
|
expect: "" + Math.PI / 6
|
|
}, {
|
|
set: {re: 99, im: 50},
|
|
fn: "conjugate",
|
|
expect: "99 - 50i"
|
|
}, {
|
|
set: "2 + 8i",
|
|
fn: "div",
|
|
param: new Complex(1, 2),
|
|
expect: "3.6 + 0.8i"
|
|
}, {
|
|
set: -Infinity,
|
|
fn: "div",
|
|
param: 3,
|
|
expect: "-Infinity"
|
|
}, {
|
|
set: {re: 1, im: 2},
|
|
fn: "add",
|
|
param: "4 + 6i",
|
|
expect: "5 + 8i"
|
|
}, {
|
|
set: {re: 5, im: 8},
|
|
fn: "sub",
|
|
param: "4 + 6i",
|
|
expect: "1 + 2i"
|
|
}, {
|
|
set: "1 + 2i",
|
|
fn: "pow",
|
|
param: "2",
|
|
expect: "-2.999999999999999 + 4.000000000000001i"
|
|
}, {
|
|
set: "1 + 2i",
|
|
fn: "pow",
|
|
param: "1 + 2i",
|
|
expect: "-0.22251715680177267 + 0.10070913113607541i"
|
|
}, {
|
|
set: {re: 1, im: 2},
|
|
fn: "pow",
|
|
param: new Complex(3, 4),
|
|
expect: "0.12900959407446697 + 0.033924092905170025i"
|
|
}, {
|
|
set: "i",
|
|
fn: "pow",
|
|
param: 7,
|
|
expect: "-i"
|
|
}, {
|
|
set: "i",
|
|
fn: "pow",
|
|
param: 4,
|
|
expect: "1"
|
|
}, {
|
|
set: "i",
|
|
fn: "pow",
|
|
param: 5,
|
|
expect: "i"
|
|
}, {
|
|
set: "0-0i",
|
|
fn: "pow",
|
|
param: 2,
|
|
expect: "0"
|
|
}, {
|
|
set: "0-0i",
|
|
fn: "pow",
|
|
param: 0,
|
|
expect: "0"
|
|
}, {
|
|
set: "1 + 4i",
|
|
fn: "sqrt",
|
|
expect: "1.600485180440241 + 1.2496210676876531i"
|
|
}, {
|
|
set: {re: -3, im: 4},
|
|
fn: "sqrt",
|
|
expect: "1 + 2i"
|
|
}, {
|
|
set: {re: 3, im: -4},
|
|
fn: "sqrt",
|
|
expect: "2 - i"
|
|
}, {
|
|
set: {re: -3, im: -4},
|
|
fn: "sqrt",
|
|
expect: "1 - 2i"
|
|
}, {
|
|
set: -Complex.E.pow(2),
|
|
fn: "log",
|
|
expect: "2 + 3.141592653589793i"
|
|
}, {
|
|
set: "4 + 3i",
|
|
fn: "log",
|
|
expect: "1.6094379124341003 + 0.6435011087932844i"
|
|
}, {
|
|
set: "4 + 3i",
|
|
fn: "exp",
|
|
expect: "-54.05175886107815 + 7.7048913727311525i"
|
|
}, {
|
|
set: "1-2i",
|
|
fn: "sqrt",
|
|
expect: "1.272019649514069 - 0.7861513777574233i"
|
|
}, {
|
|
set: {re: 1, im: 2},
|
|
fn: "sin",
|
|
expect: "3.1657785132161678 + 1.9596010414216063i"
|
|
}, {
|
|
set: "i",
|
|
fn: "cos",
|
|
expect: "1.5430806348152437"
|
|
}, {
|
|
set: "i",
|
|
fn: "acos",
|
|
expect: "1.5707963267948966 - 0.8813735870195428i"
|
|
}, {
|
|
set: {re: 1, im: 2},
|
|
fn: "cos",
|
|
expect: "2.0327230070196656 - 3.0518977991518i"
|
|
}, {
|
|
set: {re: 1, im: 2},
|
|
fn: "tan",
|
|
expect: "0.0338128260798967 + 1.0147936161466335i"
|
|
}, {
|
|
set: {re: 1, im: 3},
|
|
fn: "sinh",
|
|
expect: "-1.1634403637032504 + 0.21775955162215221i"
|
|
}, {
|
|
set: {re: 1, im: 3},
|
|
fn: "cosh",
|
|
expect: "-1.5276382501165433 + 0.1658444019189788i"
|
|
}, {
|
|
set: {re: 1, im: 3},
|
|
fn: "tanh",
|
|
expect: "0.7680176472869114 - 0.05916853956605073i"
|
|
}, {
|
|
set: {re: 1, im: 3},
|
|
fn: "inverse",
|
|
expect: "0.1 - 0.3i"
|
|
}, {
|
|
set: {re: 0.5, im: -0.5},
|
|
fn: "inverse",
|
|
expect: "1 + i"
|
|
}, {
|
|
set: "1 + i",
|
|
fn: "inverse",
|
|
expect: "0.5 - 0.5i"
|
|
}, {
|
|
set: "0",
|
|
fn: "inverse",
|
|
expect: "0"
|
|
}, {
|
|
set: Complex['EPSILON'],
|
|
fn: "equals",
|
|
param: 1e-16,
|
|
expect: "true"
|
|
}, {
|
|
set: 0,
|
|
fn: "equals",
|
|
param: "5i",
|
|
expect: "false"
|
|
}, {
|
|
set: 5,
|
|
fn: "equals",
|
|
param: "5i",
|
|
expect: "false"
|
|
}, {
|
|
set: 5,
|
|
fn: "equals",
|
|
param: 5,
|
|
expect: "true"
|
|
}, {
|
|
set: "10i",
|
|
fn: "equals",
|
|
param: "10i",
|
|
expect: "true"
|
|
}, {
|
|
set: "2 + 3i",
|
|
fn: "equals",
|
|
param: "2 + 3i",
|
|
expect: "true"
|
|
}, {
|
|
set: "2 + 3i",
|
|
fn: "equals",
|
|
param: "5i",
|
|
expect: "false"
|
|
}, {
|
|
set: "2 + 3i",
|
|
fn: "round",
|
|
param: "0",
|
|
expect: "2 + 3i"
|
|
}, {
|
|
set: "2.5 + 3.5i",
|
|
fn: "round",
|
|
param: "1",
|
|
expect: "2.5 + 3.5i"
|
|
}, {
|
|
set: "2.5 + 3.5i",
|
|
fn: "sign",
|
|
param: null,
|
|
expect: "0.5812381937190965 + 0.813733471206735i"
|
|
}, {
|
|
set: "10 + 24i",
|
|
fn: "sign",
|
|
param: null,
|
|
expect: "0.38461538461538464 + 0.9230769230769231i"
|
|
}, {
|
|
set: "1e3i",
|
|
fn: "add",
|
|
param: "3e-3 + 1e2i",
|
|
expect: "0.003 + 1100i"
|
|
}, {
|
|
set: "3.14-4i",
|
|
fn: "coth",
|
|
expect: "0.9994481238383571 + 0.0037048958915019844i"
|
|
}, {
|
|
set: "8i-31",
|
|
fn: "cot",
|
|
expect: "1.663676829121394e-7 - 1.0000001515864905i"
|
|
}, {
|
|
set: " + 7 - i + 3i - + + + + 43 + 2i - i4 + - 33 + 65 - 1 ",
|
|
expect: "-5"
|
|
}, {
|
|
set: " + 7 - i + 3i - + + + + 43 + 2i - i4 + - 33 + 65 - 1 + ",
|
|
expect: "SyntaxError: Invalid Param"
|
|
}, {
|
|
set: "-3x + 4",
|
|
expect: "SyntaxError: Invalid Param"
|
|
}, {
|
|
set: "- + 7",
|
|
expect: "-7"
|
|
}, {
|
|
set: "4 5i",
|
|
expect: "SyntaxError: Invalid Param"
|
|
}, {
|
|
set: "-",
|
|
expect: "SyntaxError: Invalid Param"
|
|
}, {
|
|
set: "2.2e-1-3.2e-1i",
|
|
expect: "0.22 - 0.32i"
|
|
}, {
|
|
set: "2.2.",
|
|
expect: "SyntaxError: Invalid Param"
|
|
}, {
|
|
set: {r: 0, phi: 4},
|
|
expect: "0"
|
|
}, {
|
|
set: {r: 1, phi: 1},
|
|
expect: "0.5403023058681398 + 0.8414709848078965i"
|
|
}
|
|
];
|
|
|
|
describe("Complex", function() {
|
|
|
|
for (var i = 0; i < tests.length; i++) {
|
|
|
|
(function(i) {
|
|
|
|
if (tests[i].fn) {
|
|
|
|
it((tests[i].fn || "") + " " + tests[i].set + ", " + (tests[i].param || ""), function() {
|
|
try {
|
|
assert.equal(tests[i].expect, new Complex(tests[i].set)[tests[i].fn](tests[i].param).toString());
|
|
} catch (e) {
|
|
assert.equal(e.toString(), tests[i].expect.toString());
|
|
}
|
|
});
|
|
|
|
} else {
|
|
|
|
it((tests[i].fn || "") + "" + tests[i].set, function() {
|
|
try {
|
|
assert.equal(tests[i].expect, new Complex(tests[i].set).toString());
|
|
} catch (e) {
|
|
assert.equal(e.toString(), tests[i].expect.toString());
|
|
}
|
|
});
|
|
}
|
|
|
|
})(i);
|
|
}
|
|
});
|
|
|
|
describe("Complex Details", function() {
|
|
|
|
it("should work with different params", function() {
|
|
assert.equal(Complex(1, -1).toString(), "1 - i");
|
|
assert.equal(Complex(0, 0).toString(), "0");
|
|
assert.equal(Complex(0, 2).toString(), "2i");
|
|
assert.equal(Complex("3 + 4i").toString(), "3 + 4i");
|
|
assert.equal(Complex("1 + i").toString(), "1 + i");
|
|
assert.equal(Complex("i").toString(), "i");
|
|
assert.equal(Complex.I.toString(), "i");
|
|
assert.equal(Complex("3 - 4i").toString(), "3 - 4i");
|
|
assert.equal(Complex("5").toString(), "5");
|
|
assert.equal(Complex(0, -2).toString(), "-2i");
|
|
assert.equal(Complex({re: 0, im: -2}).toString(), "-2i");
|
|
});
|
|
|
|
it("Complex Combinations", function() {
|
|
|
|
var zero = Complex(0, 0), one = Complex(1, 1), two = Complex(2, 2);
|
|
|
|
assert.equal(zero.toString(), "0");
|
|
assert.equal(one.toString(), "1 + i");
|
|
assert(one.neg().equals(Complex(-1, -1)));
|
|
assert(one.conjugate().equals(Complex(1, -1)));
|
|
assert.equal(one.abs(), Math.SQRT2);
|
|
assert.equal(one.arg(), Math.PI / 4);
|
|
assert.equal(one.add(one).toString(), two.toString());
|
|
assert.equal(one.sub(one).toString(), zero.toString());
|
|
assert.equal(one.mul(2).toString(), two.toString());
|
|
assert.equal(one.mul(one).toString(), Complex(0, 2).toString());
|
|
assert.equal(one.div(2).toString(), "0.5 + 0.5i");
|
|
assert.equal(one.div(one).toString(), "1");
|
|
assert.equal(one.div(0).toString(), "Infinity + Infinityi");
|
|
assert.equal(one.exp().toString(), "1.4686939399158851 + 2.2873552871788423i");
|
|
assert.equal(one.log().toString(), "0.34657359027997264 + 0.7853981633974483i");
|
|
assert.equal(one.pow(one).toString(), "0.27395725383012104 + 0.5837007587586146i");
|
|
assert.equal(one.pow(zero).toString(), "1");
|
|
assert.equal(one.sqrt().toString(), "1.09868411346781 + 0.45508986056222733i");
|
|
assert.equal(one.sin().toString(), "1.2984575814159773 + 0.6349639147847361i");
|
|
assert.equal(one.cos().toString(), "0.8337300251311491 - 0.9888977057628651i");
|
|
assert.equal(one.tan().toString(), "0.2717525853195118 + 1.0839233273386948i");
|
|
assert.equal(one.asin().toString(), "0.6662394324925153 + 1.0612750619050355i");
|
|
assert.equal(one.acos().toString(), "0.9045568943023813 - 1.0612750619050355i");
|
|
assert.equal(one.atan().toString(), "1.0172219678978514 + 0.40235947810852507i");
|
|
|
|
assert.equal(Complex("5i + 3").log().exp().toString(), "3 + 5i")
|
|
assert.equal(Complex("-2i - 1").log().exp().toString(), "-1 - 2i")
|
|
});
|
|
|
|
it("should handle inverse trig fns", function() {
|
|
|
|
var values = [
|
|
new Complex(2.3, 1.4),
|
|
new Complex(-2.3, 1.4),
|
|
new Complex(-2.3, -1.4),
|
|
new Complex(2.3, -1.4)];
|
|
|
|
var fns = ['sin', 'cos', 'tan'];
|
|
|
|
for (var i = 0; i < values.length; i++) {
|
|
|
|
for (var j = 0; j < 3; j++) {
|
|
|
|
var a = values[i]['a' + fns[j]]()[fns[j]]();
|
|
|
|
var res = values[i];
|
|
|
|
assert(Math.abs(a.re - res.re) < 1e-12 && Math.abs(a.im - res.im) < 1e-12);
|
|
}
|
|
}
|
|
});
|
|
|
|
it("should eat it's own dog food", function() {
|
|
|
|
var a = Complex(1, -5).toString();
|
|
var b = Complex(a).toString();
|
|
var c = Complex(b).mul(a);
|
|
|
|
assert.equal(c.toString(), '-24 - 10i');
|
|
});
|
|
});
|