var assert = require('assert'),
    math = require('../../../index')();

describe('mean', function() {
  it('should return the mean value of some numbers', function() {
    assert.equal(math.mean(5), 5);
    assert.equal(math.mean(3,1), 2);
    assert.equal(math.mean(0,3), 1.5);
    assert.equal(math.mean(1,3,5,2,-5), 1.2);
    assert.equal(math.mean(0,0,0,0), 0);
  });

  it('should return the mean value for complex values', function() {
    assert.deepEqual(math.mean(math.complex(2,3), math.complex(2,1)), math.complex(2,2));
    assert.deepEqual(math.mean(math.complex(2,3), math.complex(2,5)), math.complex(2,4));
  });

  it('should return the mean value for mixed real and complex values', function() {
    assert.deepEqual(math.mean(math.complex(2,4), 4), math.complex(3,2));
    assert.deepEqual(math.mean(4, math.complex(2,4)), math.complex(3,2));
  });

  it('should return the mean value from a vector', function() {
    assert.equal(math.mean(math.matrix([1,3,5,2,-5])), 1.2);
  });

  it('should return the mean for each vector on the last dimension', function() {
    assert.deepEqual(math.mean([
      [ 2, 4],
      [ 6, 8]
    ]), 5);
    assert.deepEqual(math.mean(math.matrix([
      [ 2, 4],
      [ 6, 8]
    ])), 5);
  });

  var inputMatrix = [ // this is a 4x3x2 matrix, full test coverage
          [ [10,20], [30,40], [50,60] ],
          [ [70,80], [90,100], [110,120] ],
          [ [130,140], [150,160], [170,180] ],
          [ [190,200], [210,220], [230,240]]
        ];

  it('should return the mean value along a dimension on a matrix', function() {
    assert.deepEqual(math.mean([
        [2, 6],
        [4, 10]], 1), [4, 7]);
    assert.deepEqual(math.mean([
        [2, 6],
        [4, 10]], 0), [3, 8]);
    assert.deepEqual(math.mean(inputMatrix, 0),
      [ [100, 110], [120, 130], [140,150] ]);
    assert.deepEqual(math.mean(inputMatrix, 1),
      [ [30, 40], [90,100], [150,160], [210,220]]);
    assert.deepEqual(math.mean(inputMatrix, 2),
      [[15, 35, 55],[75,95,115],[135,155,175],[195, 215, 235]]);
  });

  it('should throw an error if called with invalid number of arguments', function() {
    assert.throws(function() {math.mean()});
  });

  it('should throw an error if called with an empty array', function() {
    assert.throws(function() {math.mean([])});
  });

});