Fix missing language filtering in rule-based engine (#23)

pemistahl · Feb 6, 2020 · b9ca94c06417a53644786a5ca9f0a71217e4ff26 · b9ca94c
1 parent cc6c9ce
commit b9ca94c06417a53644786a5ca9f0a71217e4ff26
Unified Split

Showing with 6 additions and 3 deletions.

+6 −3 src/main/kotlin/com/github/pemistahl/lingua/api/LanguageDetector.kt
diff --git a/src/main/kotlin/com/github/pemistahl/lingua/api/LanguageDetector.kt b/src/main/kotlin/com/github/pemistahl/lingua/api/LanguageDetector.kt
@@ -229,10 +229,13 @@ class LanguageDetector internal constructor(
             .filter { it.value >= minimalRequiredCharCount }
             .count() > 0
 
-        return if (languagesWithMinimumRequiredCharCountExist)
-            languageCharCounts.toList().maxBy { it.second }!!.first
-        else
+        return if (languagesWithMinimumRequiredCharCountExist) {
+            val language = languageCharCounts.toList().maxBy { it.second }!!.first
+            if (language in languages) language else UNKNOWN
+        }
+        else {
             UNKNOWN
+        }
     }
 
     internal fun filterLanguagesByRules(words: List<String>): Sequence<Language> {