X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=letter_compress.py;h=42f06dae79a8357bcffd0730194d5c2d1d352f11;hb=d7d135be2f9dbb2d8398d7258f8bac24b4205a2b;hp=b5d326471e8c9412133e94e07737a7d91496dc41;hpb=eb9e6df32ed696158bf34dba6464277b648f5c74;p=python_utils.git diff --git a/letter_compress.py b/letter_compress.py index b5d3264..42f06da 100644 --- a/letter_compress.py +++ b/letter_compress.py @@ -1,5 +1,7 @@ #!/usr/bin/env python3 +"""A simple compression helper for lowercase ascii text.""" + import bitstring from collect.bidict import BiDict @@ -32,9 +34,9 @@ def compress(uncompressed: str) -> bytes: """ compressed = bitstring.BitArray() - for (n, letter) in enumerate(uncompressed): + for letter in uncompressed: if 'a' <= letter <= 'z': - bits = ord(letter) - ord('a') + 1 # 1..26 + bits = ord(letter) - ord('a') + 1 # 1..26 else: if letter not in special_characters: raise Exception(f'"{uncompressed}" contains uncompressable char="{letter}"') @@ -100,4 +102,5 @@ def decompress(kompressed: bytes) -> str: if __name__ == '__main__': import doctest + doctest.testmod()