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)))
|
||
|
|
})
|
||
|
|
})
|