Add more tests.

trekhleb · Jun 8, 2018 · 9de6bc7 · 9de6bc7
1 parent d5a8b1e
commit 9de6bc7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/algorithms/math/integer-partition/__test__/integerPartition.test.js b/src/algorithms/math/integer-partition/__test__/integerPartition.test.js
@@ -6,6 +6,9 @@ describe('integerPartition', () => {
     expect(integerPartition(2)).toBe(2);
     expect(integerPartition(3)).toBe(3);
     expect(integerPartition(4)).toBe(5);
+    expect(integerPartition(5)).toBe(7);
+    expect(integerPartition(6)).toBe(11);
+    expect(integerPartition(7)).toBe(15);
     expect(integerPartition(8)).toBe(22);
   });
 });
diff --git a/src/algorithms/math/integer-partition/integerPartition.js b/src/algorithms/math/integer-partition/integerPartition.js
@@ -1,5 +1,6 @@
 /**
- * @param {Number} number
+ * @param {number} number
+ * @return {number}
  */
 export default function integerPartition(number) {
   // Create partition matrix for solving this task using Dynamic Programming.