j6/test/matrixTest.js

Javascript scientific library (like R, NumPy, Matlab)

/* eslint-disable no-undef */
// var assert = require('chai').assert
var j6 = require('../lib/j6')
var assert = require('assert')
var eq = assert.equal
// var numeric = require('numeric')

describe('Matrix', function () {
  var A = [[1, 2, 3], [4, 5, 6], [7, 3, 9]]
  var I = j6.M.identity(3)
  describe('test tr(), eq(), all()', function () {
    it('A=A.tr().tr()', function () {
      eq(true, A.tr().tr().eq(A).all())
    })
  })
  describe('test inv(), dot(), sub(), lt()', function () {
    it('A-1*A=I', function () {
      eq(true, A.inv().dot(A).sub(I).abs().sum() < 0.001)
    })
  })
  describe('test eig()', function () {
    it('E,λ=A.eig(); E*λ.diag()-A=0', function () {
      var Eλ = j6.M.eig(A)
      var E = Eλ.E
      var λ = Eλ.lambda
      eq(true, E.dot(λ.diag()).dot(E.inv()).sub(A).abs().sum() < 0.001)
    })
  })
})