chore: Fix checksum algo for estonia

mdomke · May 9, 2024 · 6c1188e · 6c1188e
1 parent 24f4b2c
commit 6c1188e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/schwifty/checksum/estonia.py b/schwifty/checksum/estonia.py
@@ -17,4 +17,6 @@ class DefaultAlgorithm(checksum.Algorithm):
 
     def compute(self, components: list[str]) -> str:
         weights = cycle([7, 3, 1])
-        return str(10 - checksum.weighted(reversed("".join(components)), 10, weights))
+        digit = checksum.weighted(reversed("".join(components)), 10, weights)
+        digit = digit if digit == 0 else 10 - digit
+        return str(digit)