Marmelator/aho-corasick
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Issue #26 tokens report if they are 100% whitespace

Browse Source
This commit is contained in:
robert-bor 2015-09-27 20:56:04 +02:00
parent 438e546245
commit 3393e4f51f
3 changed files with 60 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/org/ahocorasick/trie/FragmentToken.java
View File

@ -2,8 +2,16 @@ package org.ahocorasick.trie;
public class FragmentToken extends Token { public class FragmentToken extends Token {
private boolean whiteSpace;
public FragmentToken(String fragment) { public FragmentToken(String fragment) {
super(fragment); super(fragment);
this.whiteSpace = true;
for (int position = 0; position < fragment.length(); position++) {
if (!Character.isWhitespace(fragment.charAt(position))) {
whiteSpace = false;
}
}
} }
@Override @Override
@ -15,4 +23,10 @@ public class FragmentToken extends Token {
public Emit getEmit() { public Emit getEmit() {
return null; return null;
} }
@Override
public boolean isWhiteSpace() {
return whiteSpace;
}
} }

4
src/main/java/org/ahocorasick/trie/Token.java
View File

@ -18,6 +18,10 @@ public abstract class Token {
return false; return false;
} }
public boolean isWhiteSpace() {
return false;
}
public abstract Emit getEmit(); public abstract Emit getEmit();
} }

56
src/test/java/org/ahocorasick/trie/TrieTest.java
View File

@ -53,6 +53,18 @@ public class TrieTest {
checkEmit(firstMatch, 1, 3, "abc"); checkEmit(firstMatch, 1, 3, "abc");
} }
@Test
public void sameKeywordTwice() {
Trie trie = Trie.builder()
.addKeyword("abc")
.addKeyword("abc")
.build();
Collection<Emit> emits = trie.parseText("abc");
assertEquals(1, emits.size());
Iterator<Emit> iterator = emits.iterator();
checkEmit(iterator.next(), 0, 2, "abc");
}
@Test @Test
public void variousKeywordsOneMatch() { public void variousKeywordsOneMatch() {
Trie trie = Trie.builder() Trie trie = Trie.builder()
@ -341,12 +353,27 @@ public class TrieTest {
Collection<Token> tokens = trie.tokenize("Alpha AlphaAlpha Alpha"); Collection<Token> tokens = trie.tokenize("Alpha AlphaAlpha Alpha");
assertEquals(6, tokens.size()); assertEquals(6, tokens.size());
Iterator<Token> tokensIt = tokens.iterator(); Iterator<Token> tokensIt = tokens.iterator();
assertToken(tokensIt.next(), "Alpha", true, true); assertToken(tokensIt.next(), "Alpha", true, true, false);
assertToken(tokensIt.next(), " ", false, false); assertToken(tokensIt.next(), " ", false, false, true);
assertToken(tokensIt.next(), "Alpha", true, false); assertToken(tokensIt.next(), "Alpha", true, false, false);
assertToken(tokensIt.next(), "Alpha", true, false); assertToken(tokensIt.next(), "Alpha", true, false, false);
assertToken(tokensIt.next(), " ", false, false); assertToken(tokensIt.next(), " ", false, false, true);
assertToken(tokensIt.next(), "Alpha", true, true); assertToken(tokensIt.next(), "Alpha", true, true, false);
}
@Test
public void whiteSpaceTokens() {
Trie trie = Trie.builder()
.addKeyword("Alpha")
.build();
Collection<Token> tokens = trie.tokenize("Alpha \tthe\t Alpha\n Alpha");
assertEquals(5, tokens.size());
Iterator<Token> tokensIt = tokens.iterator();
assertToken(tokensIt.next(), "Alpha", true, true, false);
assertToken(tokensIt.next(), " \tthe\t ", false, false, false);
assertToken(tokensIt.next(), "Alpha", true, true, false);
assertToken(tokensIt.next(), "\n ", false, false, true);
assertToken(tokensIt.next(), "Alpha", true, true, false);
} }
@Test @Test
@ -359,13 +386,13 @@ public class TrieTest {
Collection<Token> tokens = trie.tokenize("Hear: Alpha team first, Beta from the rear, Gamma in reserve"); Collection<Token> tokens = trie.tokenize("Hear: Alpha team first, Beta from the rear, Gamma in reserve");
assertEquals(7, tokens.size()); assertEquals(7, tokens.size());
Iterator<Token> tokensIt = tokens.iterator(); Iterator<Token> tokensIt = tokens.iterator();
assertToken(tokensIt.next(), "Hear: ", false, false); assertToken(tokensIt.next(), "Hear: ", false, false, false);
assertToken(tokensIt.next(), "Alpha", true, true); assertToken(tokensIt.next(), "Alpha", true, true, false);
assertToken(tokensIt.next(), " team first, ", false, false); assertToken(tokensIt.next(), " team first, ", false, false, false);
assertToken(tokensIt.next(), "Beta", true, true); assertToken(tokensIt.next(), "Beta", true, true, false);
assertToken(tokensIt.next(), " from the rear, ", false, false); assertToken(tokensIt.next(), " from the rear, ", false, false, false);
assertToken(tokensIt.next(), "Gamma", true, true); assertToken(tokensIt.next(), "Gamma", true, true, false);
assertToken(tokensIt.next(), " in reserve", false, false); assertToken(tokensIt.next(), " in reserve", false, false, false);
} }
@Test @Test
@ -473,10 +500,11 @@ public class TrieTest {
checkEmit(emits.iterator().next(), 0, 9, "#sugar-123"); checkEmit(emits.iterator().next(), 0, 9, "#sugar-123");
} }
private void assertToken(Token token, String fragment, boolean match, boolean wholeWord) { private void assertToken(Token token, String fragment, boolean match, boolean wholeWord, boolean whiteSpace) {
assertEquals(fragment, token.getFragment()); assertEquals(fragment, token.getFragment());
assertEquals(match, token.isMatch()); assertEquals(match, token.isMatch());
assertEquals(wholeWord, token.isWholeWord()); assertEquals(wholeWord, token.isWholeWord());
assertEquals(whiteSpace, token.isWhiteSpace());
} }
private void checkEmit(Emit next, int expectedStart, int expectedEnd, String expectedKeyword) { private void checkEmit(Emit next, int expectedStart, int expectedEnd, String expectedKeyword) {