From 9de6bc7de3aeb1421e8947d80d28a1a9bb8253c2 Mon Sep 17 00:00:00 2001
From: Oleksii Trekhleb <trehleb@gmail.com>
Date: Fri, 8 Jun 2018 16:54:07 +0300
Subject: [PATCH] Add more tests.

---
 .../math/integer-partition/__test__/integerPartition.test.js   | 3 +++
 src/algorithms/math/integer-partition/integerPartition.js      | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/algorithms/math/integer-partition/__test__/integerPartition.test.js b/src/algorithms/math/integer-partition/__test__/integerPartition.test.js
index affcbca6..e78b1dfd 100644
--- 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
index b77cc0a3..283d9a40 100644
--- 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.