mathjs/test/function/matrix/zeros.test.js

// test zeros
var assert = require('assert');
var math = require('../../../index.js'),
    zeros = math.zeros,
    matrix = math.matrix;

describe('zeros', function() {

  it('should create a scalar', function () {
    assert.deepEqual(zeros(), 0);
    assert.deepEqual(zeros([]), 0);
  });

  it('should create a vector with zeros', function () {
    assert.deepEqual(zeros(3), matrix([0,0,0]));
    assert.deepEqual(zeros(matrix([4])), matrix([0,0,0,0]));
    assert.deepEqual(zeros([4]), [0,0,0,0]);
    assert.deepEqual(zeros(0), matrix([]));
  });

  it('should create a 2D matrix with zeros from an array', function () {
    assert.deepEqual(zeros(2,3), matrix([[0,0,0],[0,0,0]]));
    assert.deepEqual(zeros(3,2), matrix([[0,0],[0,0],[0,0]]));
    assert.deepEqual(zeros([3,2]), [[0,0],[0,0],[0,0]]);
  });

  it('should create a matrix with zeros from a matrix', function () {
    assert.deepEqual(zeros(matrix([3])), matrix([0,0,0]));
    assert.deepEqual(zeros(matrix([3,2])), matrix([[0,0],[0,0],[0,0]]));

    // TODO: do we want to support the following? maybe better not
    assert.deepEqual(zeros(matrix([[[3]],[[2]]])), matrix([[0,0],[0,0],[0,0]]));
  });

  it('should create a 3D matrix with zeros', function () {
    var res = [
      [
        [0,0,0,0],
        [0,0,0,0],
        [0,0,0,0]
      ],
      [
        [0,0,0,0],
        [0,0,0,0],
        [0,0,0,0]
      ]
    ];

    assert.deepEqual(zeros(2,3,4), matrix(res));
    assert.deepEqual(zeros(matrix([2,3,4])), matrix(res));
    assert.deepEqual(zeros([2,3,4]), res);
  });

  // TODO: test option math.options.matrix.defaultType

  it('should create a matrix with zeros with the same size as original matrix', function () {
    var a = matrix([[1, 2, 3], [4, 5, 6]]);
    assert.deepEqual(zeros(math.size(a)).size(), a.size());
  });

  // TODO: test with invalid input

});