@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)); } }