1-count-words.js

/*
  This exercise is to write function that counts the number of times each word appears in a string.

  Write a function called countWords that
    - takes a string as an argument
    - returns an object where
          - the keys are the words from the string and
          - the values are the number of times the word appears in the string

  Example
  If the function is given the input:

  "you and me and you";

  the object returned would be:

  { you: 2, and: 2, me: 1 }

  To complete this exercise you should understand
    - Strings and string manipulation
    - Loops or forEach
    - Comparison inside if statements
    - Setting values on an object
*/

<<<<<<< HEAD
function countWords(str) {
  let array = str.split(" ");
  const counts = {};
  for (let i = 0; i < array.length; i++) {
    const word = array[i];
    if (counts[word]) {
      counts[word]++;
    } else {
      counts[word] = 1;
    }
  }
  return counts;
=======
function countWords(string) {
  // write code here

  let strLower = string.toLowerCase();
  const wordCount = {};

  // split the string into an array of words
  const words = strLower.split(" ");

  // loop through each word
  words.forEach((word) => {
    // check if the word already exists in the object
    if (wordCount[word]) {
      // if it does, increment the value of that key
      wordCount[word]++;
    } else {
      // if it doesn't, add it to the object with a value of 1
      wordCount[word] = 1;
    }
  });

  return wordCount;
>>>>>>> 26cab01f55b1033f71c12f38340eba5bac196f45
}

/* ======= TESTS - DO NOT MODIFY =====
- To run the tests for this exercise, run `npm run extra-tests`
- To run all exercises/tests in the mandatory folder, run `npm test`
- (Reminder: You must have run `npm install` one time before this will work!)
*/

test("Code works for a small string", () => {
  expect(countWords("I love CodeYourFuture")).toEqual({
    I: 1,
    love: 1,
    CodeYourFuture: 1,
  });
});

test("A string with, some punctuation", () => {
  expect(countWords("A string with, some punctuation")).toEqual({
    A: 1,
    string: 1,
    "with,": 1,
    some: 1,
    punctuation: 1,
  });
});

test("Empty string", () => {
  expect(countWords("")).toEqual({});
});

test("Example task string", () => {
  expect(
    countWords(
      "you're braver than you believe, stronger than you seem, and smarter than you think"
    )
  ).toEqual({
    "you're": 1,
    and: 1,
    "believe,": 1,
    braver: 1,
    "seem,": 1,
    smarter: 1,
    stronger: 1,
    than: 3,
    think: 1,
    you: 3,
  });
});