fix(string/countWords): punctation chars

martinkr · Dec 7, 2021 · 62e1460 · 62e1460
1 parent 388bac5
commit 62e1460
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/string/count-words.js b/src/string/count-words.js
@@ -3,6 +3,7 @@
  * @param {String} str 
  * @returns {Number}
  */
-const fn = str => str.trim().split(/\s+/g).length;
+// const fn = str => str.trim().split(/\W+/g).length;
+const fn = str => str.trim().split(/\s+/g).map(i => i.replace(/[\[\]?.,\/#!$%\^&\*;:{}=\"\-_~()…–—·'’]/g,"")).filter(i=>i).length;
 
 export default fn;
diff --git a/test/string/count-words.spec.js b/test/string/count-words.spec.js
@@ -25,3 +25,13 @@ test('returns the amount of words in a string if the last chars are whitespaces'
   let input = "foo bar baz   ";
   t.is( countWords(input), 3);
 });
+
+test('returns the amount of words in a string if the there are regular punctation marks and whitespaces', t => {
+  let input = "~ …  ;,. ! ? [foo1] foo2 foo3-bar3 baz2  – — · / ( ) { } [ ] \" ' : ’ baz1";
+  t.is( countWords(input), 5);
+});
+
+test('returns the amount of words in a string if they are surounded by spaces', t => {
+  let input = "foo bar baz – — · / ( ) { } [ ] ~ … . , ; ! ? ' : ’ \" foobar [foo] ";
+  t.is( countWords(input), 5);
+});