mathjs/test/unit-tests/function/matrix/identity.test.js

75 lines
3.5 KiB
JavaScript

import assert from 'assert'
import math from '../../../../src/bundleAny'
const matrix = math.matrix
const identity = math.identity
describe('identity', function () {
it('should create an empty matrix', function () {
assert.deepStrictEqual(identity(), matrix())
assert.deepStrictEqual(identity([]), [])
assert.deepStrictEqual(identity(matrix([])), matrix())
})
it('should create an empty sparse matrix', function () {
assert.deepStrictEqual(identity('sparse'), matrix('sparse'))
assert.deepStrictEqual(identity(matrix([], 'sparse')), matrix('sparse'))
})
it('should create an identity matrix of the given size', function () {
assert.deepStrictEqual(identity(1), matrix([[1]]))
assert.deepStrictEqual(identity(2), matrix([[1, 0], [0, 1]]))
assert.deepStrictEqual(identity([2]), [[1, 0], [0, 1]])
assert.deepStrictEqual(identity(2, 3), matrix([[1, 0, 0], [0, 1, 0]]))
assert.deepStrictEqual(identity(3, 2), matrix([[1, 0], [0, 1], [0, 0]]))
assert.deepStrictEqual(identity([3, 2]), [[1, 0], [0, 1], [0, 0]])
assert.deepStrictEqual(identity(math.matrix([3, 2])), matrix([[1, 0], [0, 1], [0, 0]]))
assert.deepStrictEqual(identity(3, 3), matrix([[1, 0, 0], [0, 1, 0], [0, 0, 1]]))
})
it('should create an identity matrix with storage type css of the given size', function () {
assert.deepStrictEqual(identity(1, 'sparse'), matrix([[1]], 'sparse'))
assert.deepStrictEqual(identity(2, 'sparse'), matrix([[1, 0], [0, 1]], 'sparse'))
assert.deepStrictEqual(identity(2, 3, 'sparse'), matrix([[1, 0, 0], [0, 1, 0]], 'sparse'))
assert.deepStrictEqual(identity(3, 2, 'sparse'), matrix([[1, 0], [0, 1], [0, 0]], 'sparse'))
assert.deepStrictEqual(identity(3, 3, 'sparse'), matrix([[1, 0, 0], [0, 1, 0], [0, 0, 1]], 'sparse'))
})
it('should create an identity matrix with bignumbers', function () {
const zero = math.bignumber(0)
const one = math.bignumber(1)
const two = math.bignumber(2)
const three = math.bignumber(3)
assert.deepStrictEqual(identity(two), matrix([[one, zero], [zero, one]]))
// assert.deepStrictEqual(identity(two, 'sparse'), matrix([[one,zero],[zero,one]], 'sparse')); // FIXME: identity css
assert.deepStrictEqual(identity(two, three), matrix([[one, zero, zero], [zero, one, zero]]))
// assert.deepStrictEqual(identity(two, three, 'sparse'), matrix([[one,zero,zero],[zero,one,zero]], 'sparse')); // FIXME: identity css
})
it('should return an array when setting matrix=="array"', function () {
const math2 = math.create({ matrix: 'Array' })
assert.deepStrictEqual(math2.identity(2), [[1, 0], [0, 1]])
})
it('should throw an error with an invalid input', function () {
assert.throws(function () { identity(3, 3, 2) })
assert.throws(function () { identity([3, 3, 2]) })
assert.throws(function () { identity([3, 3], 2) })
assert.throws(function () { identity([3.2, 3]) })
assert.throws(function () { identity([3, 3.2]) })
assert.throws(function () { identity([3.2, 3.2]) })
assert.throws(function () { identity([2, 'str']) })
assert.throws(function () { identity(['str', 2]) })
assert.throws(function () { identity([-2, 2]) })
assert.throws(function () { identity([2, -2]) })
})
it('should LaTeX identity', function () {
const expression = math.parse('identity(2)')
assert.strictEqual(expression.toTex(), '\\mathrm{identity}\\left(2\\right)')
})
it('should throw an error when using the deprecated function eye', function () {
assert.throws(function () { math.eye(2) }, /Function "eye" is renamed to "identity"/)
})
})