X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=letter_compress.py;h=9b4cf194c3d97a83c68c47bd0199a65c3a6e1698;hb=36fea7f15ed17150691b5b3ead75450e575229ef;hp=b5d326471e8c9412133e94e07737a7d91496dc41;hpb=eb9e6df32ed696158bf34dba6464277b648f5c74;p=python_utils.git diff --git a/letter_compress.py b/letter_compress.py index b5d3264..9b4cf19 100644 --- a/letter_compress.py +++ b/letter_compress.py @@ -34,10 +34,12 @@ def compress(uncompressed: str) -> bytes: compressed = bitstring.BitArray() for (n, letter) in enumerate(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}"') + raise Exception( + f'"{uncompressed}" contains uncompressable char="{letter}"' + ) bits = special_characters[letter] compressed.append(f"uint:5={bits}") while len(compressed) % 8 != 0: @@ -100,4 +102,5 @@ def decompress(kompressed: bytes) -> str: if __name__ == '__main__': import doctest + doctest.testmod()