Merge pull request #138 from qualitymanifest/tests/trie-unique-word-count

Add unit tests for Trie/unique-word-count

Author: TheSTL

src/_DataStructures_/Trees/Trie/unique-word-count/index.js

@@ -1,6 +1,3 @@
-/* eslint-disable no-unused-vars */
-const Trie = require('../index');
-
 function uniqueWordCount(root) {
   let result = 0;
   if (root.isEndOfWord) {
@@ -14,10 +11,4 @@ function uniqueWordCount(root) {
   return result;
 }
 
-// const words = ['bed', 'ball', 'apple', 'java', 'javascript', 'bed'];
-// const trie = new Trie();
-
-// words.forEach(word => trie.insert(word));
-// console.log(uniqueWordCount(trie.root));
-
 module.exports = uniqueWordCount;

src/_DataStructures_/Trees/Trie/unique-word-count/index.test.js

@@ -0,0 +1,26 @@
+const Trie = require('../index');
+const uniqueWordCount = require('.');
+
+describe('Trie Unique Word Count', () => {
+  it('counts an empty trie', () => {
+    const trie = new Trie();
+    const wordCount = uniqueWordCount(trie.root);
+    expect(wordCount).toEqual(0);
+  });
+
+  it('counts unique words', () => {
+    const trie = new Trie();
+    const words = ['one', 'two', 'three', 'four'];
+    words.forEach(word => trie.insert(word));
+    const wordCount = uniqueWordCount(trie.root);
+    expect(wordCount).toEqual(4);
+  });
+
+  it('does not count duplicate words', () => {
+    const trie = new Trie();
+    const words = ['one', 'one', 'two', 'three'];
+    words.forEach(word => trie.insert(word));
+    const wordCount = uniqueWordCount(trie.root);
+    expect(wordCount).toEqual(3);
+  });
+});