Fix memoize with more than 1 argument

ivanzusko · Jun 1, 2017 · c5ddc1f · c5ddc1f
1 parent 4455bc8
commit c5ddc1f
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/src/memoize.js b/src/memoize.js
@@ -29,4 +29,6 @@ var toString = require('./toString');
  *      factorial(5); //=> 120
  *      count; //=> 1
  */
-module.exports = memoizeWith(toString);
+module.exports = memoizeWith(function() {
+  return toString(arguments);
+});
diff --git a/test/memoize.js b/test/memoize.js
@@ -35,4 +35,21 @@ describe('memoize', function() {
     eq(count, 1);
   });
 
+  it('works with 2 arguments', function() {
+    var called = 0;
+
+    var add = R.memoize(function(x, y) {
+      called += 1;
+      return x + y;
+    });
+
+    eq(add(1, 2), 3);
+    eq(called, 1);
+
+    eq(add(1, 2), 3);
+    eq(called, 1);
+
+    eq(add(1, 3), 4);
+    eq(called, 2);
+  });
 });