public void testWordFrequency() { HashMap<String, Integer> expected = new HashMap<String, Integer>(); expected.put("one", 1); expected.put("two", 1); expected.put("three", 1); assertEquals(expected, Lexer.wordFrequency("One two three")); expected.put("three", 2); assertEquals(expected, Lexer.wordFrequency("One two three three")); }
public void testWordFrequencyIgnoreSpecialChars() { Character[] replace = {'.', ',', '!', '?'}; HashMap<String, Integer> expected = new HashMap<String, Integer>(); expected.put("one", 1); expected.put("two", 1); expected.put("three", 1); assertEquals(expected, Lexer.wordFrequency("one. two, three!?", replace)); }