improved negativeWords Set check

karronoli · Jul 18, 2020 · cf5336a · cf5336a
1 parent 93a3845
commit cf5336a
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/redpen-core/src/main/java/cc/redpen/validator/sentence/DoubleNegativeValidator.java b/redpen-core/src/main/java/cc/redpen/validator/sentence/DoubleNegativeValidator.java
@@ -55,15 +55,13 @@ public void validate(Sentence sentence) {
         String errorPart = "";
         List<TokenElement> tokens = sentence.getTokens();
         for (TokenElement token : tokens) {
-            for (String negativeWord : negativeWords) {
-                if (token.getSurface().toLowerCase().equals(negativeWord)) {
-                    count++;
-                    errorPart += " " + token.getSurface().toLowerCase();
-                }
-                if (count >= 2) {
-                    addLocalizedError(sentence, errorPart); 
-                    return;
-                }
+            if (negativeWords.contains(token.getSurface().toLowerCase())) {
+                count++;
+                errorPart += " " + token.getSurface().toLowerCase();
+            }
+            if (count >= 2) {
+                addLocalizedError(sentence, errorPart); 
+                return;
             }
         }
     }