@Test
  public void testMapByPartOfSpeech() {
    Map<PartOfSpeechType, ArrayList<Word>> partOfSpeechWordMap =
        BasicWordMapDao.mapByPartOfSpeech(wordsToReturn);

    assertEquals(10, partOfSpeechWordMap.size());

    for (Word word : wordsToReturn) {
      assertTrue(partOfSpeechWordMap.get(word.getPartOfSpeech()).contains(word));
    }
  }