ndarray-lu-solve/test/2var_lu_arg.js

solve a linear system of equations from an LU decomposition

var test = require('tape');
var solve = require('../');
var crout = require('ndarray-crout-decomposition');
var ndarray = require('ndarray');
var zeros = require('zeros');

var size = 2;
var A = ndarray(
    [ 8, 1, 1, 12, 5, 5 ],
    [ size + 1, size ], [ 1, size + 1 ]
);
var LU = zeros([ size, size ]);
var expected = [ 0, 1 ];

test('2-variable system, in-place LU', function (t) {
    crout(A.hi(size,size), LU);
    
    var X = new Float64Array(size);
    var Y = new Float64Array(size);
    var solution = solve(LU, A.lo(size,0).pick(0), X, Y);
    
    t.equal(X, solution);
    t.deepEqual([].slice.call(X), expected);
    
    for (var y = 0; y < size; y++) {
        var c = 0;
        for (var i = 0; i < size; i++) {
            c += X[i] * A.get(i, y);
        }
        t.equal(c, A.get(size, y));
    }
    t.end();
});