97 lines
3.3 KiB
JavaScript

const assert = require('assert')
const math = require('../../../src/main')
const BigNumber = math.type.BigNumber
const Complex = math.type.Complex
const DenseMatrix = math.type.DenseMatrix
const Unit = math.type.Unit
const sum = math.sum
describe('sum', function () {
it('should return the sum of numbers', function () {
assert.equal(sum(5), 5)
assert.equal(sum(3, 1), 4)
assert.equal(sum(1, 3), 4)
assert.equal(sum(1, 3, 5, 2), 11)
assert.equal(sum(0, 0, 0, 0), 0)
})
it('should return the sum of big numbers', function () {
assert.deepEqual(sum(new BigNumber(1), new BigNumber(3), new BigNumber(5), new BigNumber(2)),
new BigNumber(11))
})
it('should return the sum of strings (convert them to numbers)', function () {
assert.strictEqual(sum('2', '3', '4', '5'), 14)
assert.strictEqual(sum([['2', '3'], ['4', '5']]), 14)
})
it('should return the sum of complex numbers', function () {
assert.deepEqual(sum(new Complex(2, 3), new Complex(-1, 2)), new Complex(1, 5))
})
it('should return the sum of mixed numbers and complex numbers', function () {
assert.deepEqual(sum(2, new Complex(-1, 3)), new Complex(1, 3))
})
it('should return the sum from an array', function () {
assert.equal(sum([1, 3, 5, 2, -5]), 6)
})
it('should return the sum of units', function () {
assert.deepEqual(sum([new Unit(5, 'mm'), new Unit(10, 'mm'), new Unit(15, 'mm')]), new Unit(30, 'mm'))
})
it('should return the sum from an 1d matrix', function () {
assert.equal(sum(new DenseMatrix([1, 3, 5, 2, -5])), 6)
})
it('should return the sum element from a 2d array', function () {
assert.deepEqual(sum([
[1, 4, 7],
[3, 0, 5],
[-1, 11, 9]
]), 39)
})
it('should return the sum element from a 2d matrix', function () {
assert.deepEqual(sum(new DenseMatrix([
[1, 4, 7],
[3, 0, 5],
[-1, 11, 9]
])), 39)
})
it('should throw an error if called with invalid number of arguments', function () {
assert.throws(function () { sum() })
})
it('should throw an error if called with not yet supported argument dim', function () {
assert.throws(function () { sum([], 2) }, /not yet supported/)
})
it('should return zero if called with an empty array', function () {
const bigMath = math.create({number: 'BigNumber'})
const fracMath = math.create({number: 'Fraction'})
const big = bigMath.sum([])
const frac = fracMath.sum([])
assert.equal(sum([]), 0)
assert.equal(big.type, 'BigNumber')
assert.equal(frac.type, 'Fraction')
assert.equal(math.equal(bigMath.sum([]), new BigNumber(0)).valueOf(), true)
assert.equal(math.equal(fracMath.sum([]), new fracMath.type.Fraction(0)), true)
})
it('should throw an error if called with invalid type of arguments', function () {
assert.throws(function () { sum(new Date(), 2) }, /Cannot calculate sum, unexpected type of argument/)
assert.throws(function () { sum(2, 3, null) }, /Cannot calculate sum, unexpected type of argument/)
assert.throws(function () { sum([2, 3, null]) }, /Cannot calculate sum, unexpected type of argument/)
})
it('should LaTeX sum', function () {
const expression = math.parse('sum(1,2,3)')
assert.equal(expression.toTex(), '\\mathrm{sum}\\left(1,2,3\\right)')
})
})