X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=letter_compress.py;h=42f06dae79a8357bcffd0730194d5c2d1d352f11;hb=f2600f30801c849fc1d139386e3ddc3c9eb43e30;hp=9a6f32db43fb09762cca71f79cb268a8862e2452;hpb=791755ededfc0de84a81b4c7313830efbd27bf8c;p=python_utils.git diff --git a/letter_compress.py b/letter_compress.py index 9a6f32d..42f06da 100644 --- a/letter_compress.py +++ b/letter_compress.py @@ -1,10 +1,12 @@ #!/usr/bin/env python3 +"""A simple compression helper for lowercase ascii text.""" + import bitstring -from collect.bidict import bidict +from collect.bidict import BiDict -special_characters = bidict( +special_characters = BiDict( { ' ': 27, '.': 28, @@ -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()