in_target_language: avoid ZeroDivisionError (fixes #18)

adbar · Oct 5, 2022 · c87066a · c87066a
1 parent 6102563
commit c87066a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/simplemma/langdetect.py b/simplemma/langdetect.py
@@ -51,7 +51,9 @@ def in_target_language(text: str, lang: Optional[Tuple[str]] = None) -> float:
             if candidate is not None:
                 in_target += 1
                 break
-    return in_target / total
+    if total > 0 and in_target > 0:
+        return in_target / total
+    return 0
 
 
 def _return_default() -> List[Tuple[str, float]]:

diff --git a/tests/test_langdetect.py b/tests/test_langdetect.py
@@ -27,6 +27,7 @@ def test_detection():
         lang=("cs", "sk"),
     ) == [("cs", 0.625), ("unk", 0.375), ("sk", 0.125)]
     # target language
+    assert in_target_language("", lang="en") == 0
     assert (
         in_target_language(
             "opera post physica posita (τὰ μετὰ τὰ φυσικά)", lang=("la",)