mathjs/test/utils/function.test.js

var assert = require('assert');
var functionUtils = require('../../lib/utils/function');

describe('util.function', function() {

  describe('memoize', function () {

    it('should memoize a function with one argument', function () {
      var f = function (x) {return x * x};

      var m = functionUtils.memoize(f);

      assert.strictEqual(m(2), 4);
      assert.strictEqual(m(3), 9);
    });

    it('should memoize a function with two arguments', function () {
      var f = function (x, y) {return x * y};

      var m = functionUtils.memoize(f);

      assert.strictEqual(m(2, 3), 6);

      // hash should differ
      assert.strictEqual(m(1, 23), 23);
      assert.strictEqual(m(12, 3), 36);
    });

    it('should memoize a function with objects as arguments', function () {
      var f = function (obj) {return obj.x * obj.y};

      var m = functionUtils.memoize(f);

      assert.strictEqual(m({x: 2, y: 3}), 6);
      assert.deepEqual(Object.keys(m.cache), ['[{"x":2,"y":3}]']);
      assert.strictEqual(m.cache['[{"x":2,"y":3}]'], 6);
    });

    it('should memoize a function with a custom hashIt function', function () {
      var f = function (obj) {return obj.id};
      var hashIt = function (args) {
        return 'id:' + args[0].id;
      };

      var m = functionUtils.memoize(f, hashIt);

      assert.strictEqual(m({id: 2}), 2);
      assert.deepEqual(Object.keys(m.cache), ['id:2']);
      assert.strictEqual(m.cache['id:2'], 2);
    });

    it('should really return the cached result', function () {
      var a = 2;
      var f = function (x) {return a}; // trick: no pure function

      var m = functionUtils.memoize(f);

      assert.strictEqual(m(4), 2);
      a = 3;
      assert.strictEqual(m(4), 2);
    });

  });

  describe ('maxArgumentCount', function () {

    it('should calculate the max argument count of a typed function', function() {
      var a = function () {};
      a.signatures = {
        'number, number': function () {},
        'number': function () {}
      };
      assert.equal(functionUtils.maxArgumentCount(a), 2);

      var b = function () {};
      b.signatures = {
        'number': function () {},
        'number, number': function () {}
      };
      assert.equal(functionUtils.maxArgumentCount(b), 2);

      var c = function () {};
      c.signatures = {
        'number': function () {},
        'BigNumber': function () {}
      };
      assert.equal(functionUtils.maxArgumentCount(c), 1);

      var d = function () {};
      d.signatures = {
        'number,number': function () {},
        'number': function () {},
        'number,any,number': function () {}
      };
      assert.equal(functionUtils.maxArgumentCount(d), 3);
    });

    it('should return -1 for regular functions', function() {
      assert.equal(functionUtils.maxArgumentCount(function () {}), -1);
    });

  });

});