Glen Whitney
0dbb95bbbe
Implements a simply polynomial root finder function polynomialRoot, intended to be used for benchmarking against mathjs. For this purpose, adds numerous other functions (e.g. cbrt, arg, cis), refactors sqrt (so that you can definitely get the complex square root when you want it), and makes numerous enhancements to the core so that a template can match after conversions. Co-authored-by: Glen Whitney <glen@studioinfinity.org> Reviewed-on: #57
64 lines
2.6 KiB
JavaScript
64 lines
2.6 KiB
JavaScript
import assert from 'assert'
|
|
import * as approx from '../../tools/approx.mjs'
|
|
import math from '../../src/pocomath.mjs'
|
|
|
|
describe('polynomialRoot', () => {
|
|
it('should solve a linear equation with real coefficients', function () {
|
|
assert.deepEqual(math.polynomialRoot(6, 3), math.tuple(-2))
|
|
assert.deepEqual(
|
|
math.polynomialRoot(math.complex(-3, 2), 2),
|
|
math.tuple(math.complex(1.5, -1)))
|
|
assert.deepEqual(
|
|
math.polynomialRoot(math.complex(3, 1), math.complex(-1, -1)),
|
|
math.tuple(math.complex(2, -1)))
|
|
})
|
|
// Should be safe now to capture the functions:
|
|
const complex = math.complex
|
|
const pRoot = math.polynomialRoot
|
|
const tup = math.tuple
|
|
it('should solve a quadratic equation with a double root', function () {
|
|
assert.deepEqual(pRoot(4, 4, 1), tup(-2))
|
|
assert.deepEqual(
|
|
pRoot(complex(0, 2), complex(2, 2), 1), tup(complex(-1, -1)))
|
|
})
|
|
it('should solve a quadratic with two distinct roots', function () {
|
|
assert.deepEqual(pRoot(-3, 2, 1), tup(1, -3))
|
|
assert.deepEqual(pRoot(-2, 0, 1), tup(math.sqrt(2), -math.sqrt(2)))
|
|
assert.deepEqual(
|
|
pRoot(4, 2, 1),
|
|
tup(complex(-1, math.sqrt(3)), complex(-1, -math.sqrt(3))))
|
|
assert.deepEqual(
|
|
pRoot(complex(3, 1), -3, 1), tup(complex(1, 1), complex(2, -1)))
|
|
})
|
|
it('should solve a cubic with a triple root', function () {
|
|
assert.deepEqual(pRoot(8, 12, 6, 1), tup(-2))
|
|
assert.deepEqual(
|
|
pRoot(complex(-2, 11), complex(9, -12), complex(-6, 3), 1),
|
|
tup(complex(2, -1)))
|
|
})
|
|
it('should solve a cubic with one simple and one double root', function () {
|
|
assert.deepEqual(pRoot(4, 0, -3, 1), tup(-1, 2))
|
|
assert.deepEqual(
|
|
pRoot(complex(9, 9), complex(15, 6), complex(7, 1), 1),
|
|
tup(complex(-1, -1), -3))
|
|
assert.deepEqual(
|
|
pRoot(complex(0, 6), complex(6, 8), complex(5, 2), 1),
|
|
tup(-3, complex(-1, -1)))
|
|
assert.deepEqual(
|
|
pRoot(complex(2, 6), complex(8, 6), complex(5, 1), 1),
|
|
tup(complex(-3, 1), complex(-1, -1)))
|
|
})
|
|
it('should solve a cubic with three distinct roots', function () {
|
|
approx.deepEqual(pRoot(6, 11, 6, 1), tup(-3, -1, -2))
|
|
approx.deepEqual(
|
|
pRoot(-1, -2, 0, 1),
|
|
tup(-1, (1 + math.sqrt(5)) / 2, (1 - math.sqrt(5)) / 2))
|
|
approx.deepEqual(
|
|
pRoot(1, 1, 1, 1),
|
|
tup(-1, complex(0, -1), complex(0, 1)))
|
|
approx.deepEqual(
|
|
pRoot(complex(0, -10), complex(8, 12), complex(-6, -3), 1),
|
|
tup(complex(1, 1), complex(3, 1), complex(2, 1)))
|
|
})
|
|
})
|