mirror of
https://github.com/josdejong/mathjs.git
synced 2025-12-08 19:46:04 +00:00
72 lines
3.0 KiB
JavaScript
72 lines
3.0 KiB
JavaScript
// test sqrtm
|
|
import assert from 'assert'
|
|
|
|
import approx from '../../../../tools/approx'
|
|
import math from '../../../../src/bundleAny'
|
|
|
|
describe('sqrtm', function () {
|
|
const A = [[5, 2], [4, 7]]
|
|
const AA = [[33, 24], [48, 57]]
|
|
|
|
const B = [[1.5667, 1.7408], [2.6112, 4.1779]]
|
|
const BB = [[7, 10], [15, 22]]
|
|
|
|
it('should return the principal square root of a matrix', function () {
|
|
approx.deepEqual(math.sqrtm(AA), A)
|
|
approx.deepEqual(math.sqrtm(BB), B)
|
|
|
|
approx.deepEqual(
|
|
math.sqrtm(
|
|
[[5, -4, 1, 0, 0],
|
|
[-4, 6, -4, 1, 0],
|
|
[1, -4, 6, -4, 1],
|
|
[0, 1, -4, 6, -4],
|
|
[0, 0, 1, -4, 6]]),
|
|
[[2.001450032980806, -0.997069854672086, 0.004226841220338, 0.004648098208404, 0.003162179891248],
|
|
[-0.997069854672086, 2.006191486385745, -0.990363307725271, 0.011838782789231, 0.009369460553432],
|
|
[0.004226841220338, -0.990363307725271, 2.017072849046023, -0.974622709753106, 0.026274470491696],
|
|
[0.004648098208404, 0.011838782789231, -0.974622709753106, 2.050268428894304, -0.919971837350421],
|
|
[0.003162179891248, 0.009369460553432, 0.026274470491696, -0.919971837350421, 2.269992000979243]])
|
|
})
|
|
|
|
it('should return the principal square root of a matrix with just one value', function () {
|
|
assert.deepStrictEqual(math.sqrtm([4]), [2])
|
|
assert.deepStrictEqual(math.sqrtm([16]), [4])
|
|
assert.deepStrictEqual(math.sqrtm([20.25]), [4.5])
|
|
})
|
|
|
|
it('should return the principal square root of a matrix of big numbers', function () {
|
|
assert.deepStrictEqual(math.round(math.sqrtm(math.bignumber(AA)), 20), math.bignumber(A))
|
|
})
|
|
|
|
it('math.pow(math.sqrtm(A), 2) should equal A', function () {
|
|
approx.deepEqual(math.pow(math.sqrtm(A), 2), A)
|
|
approx.deepEqual(math.pow(math.sqrtm(B), 2), B)
|
|
approx.deepEqual(math.pow(math.sqrtm(AA), 2), AA)
|
|
approx.deepEqual(math.pow(math.sqrtm(BB), 2), BB)
|
|
})
|
|
|
|
it('should throw an error in case of non-square matrices', function () {
|
|
assert.throws(function () { math.sqrtm([1, 2, 3]) }, /Matrix must be square/)
|
|
assert.throws(function () { math.sqrtm([[1, 2, 3]]) }, /Matrix must be square/)
|
|
assert.throws(function () { math.sqrtm([[1, 2, 3], [4, 5, 6]]) }, /Matrix must be square/)
|
|
})
|
|
|
|
it('should LaTeX sqrtm', function () {
|
|
const expression = math.parse('sqrtm([[33, 24], [48, 57]])')
|
|
assert.strictEqual(expression.toTex(), '{\\begin{bmatrix}33&24\\\\48&57\\\\\\end{bmatrix}}^{\\frac{1}{2}}')
|
|
})
|
|
|
|
it('should return the result in the same format as the input', function () {
|
|
assert.strictEqual(math.typeOf(math.sqrtm(A)), 'Array')
|
|
assert.strictEqual(math.typeOf(math.sqrtm(B)), 'Array')
|
|
assert.strictEqual(math.typeOf(math.sqrtm(AA)), 'Array')
|
|
assert.strictEqual(math.typeOf(math.sqrtm(BB)), 'Array')
|
|
|
|
assert.strictEqual(math.typeOf(math.sqrtm(math.matrix(A))), 'Matrix')
|
|
assert.strictEqual(math.typeOf(math.sqrtm(math.matrix(B))), 'Matrix')
|
|
assert.strictEqual(math.typeOf(math.sqrtm(math.matrix(AA))), 'Matrix')
|
|
assert.strictEqual(math.typeOf(math.sqrtm(math.matrix(BB))), 'Matrix')
|
|
})
|
|
})
|