def test_custom_alphabet64_invalid():
CUSTOM = 'abcdefghijklmnopqrstuvwxyz0123456789.:,;*@#$%&/()[]{}=_-ABCDEFGH'
# bad custom alphabet
with pytest.raises(AssertionError):
to64(1, 'bad')
with pytest.raises(AssertionError):
to64(1, CUSTOM + '!-')
# values not in custom alphabet
with pytest.raises(ValueError):
from64('J', CUSTOM)
with pytest.raises(ValueError):
from64('!', CUSTOM)
def test_custom_alphabet64():
CUSTOM = 'abcdefghijklmnopqrstuvwxyz0123456789.:,;*@#$%&/()[]{}=_-ABCDEFGH'
assert to64(555, CUSTOM) == 'i$'
assert from64('AA', CUSTOM) == 3640
RANDOM = [
c for c in
'0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ=_'
]
random.shuffle(RANDOM)
RANDOM = ''.join(RANDOM)
RCUSTOM = [c for c in CUSTOM]
random.shuffle(RCUSTOM)
RCUSTOM = ''.join(RCUSTOM)
rnum = random.randint(0, 5555)
assert from64(to64(rnum, CUSTOM), CUSTOM) == rnum
assert from64(to64(rnum, RANDOM), RANDOM) == rnum
assert from64(to64(rnum, RCUSTOM), RCUSTOM) == rnum
def test_to64():
assert to64(0) == '0'
assert to64(10) == 'a'
assert to64(125) == '1Z'
assert to64(126) == '1='
assert to64(127) == '1_'
assert to64(128) == '20'
assert len(to64(pow(3, 1979))) == 523
with pytest.raises(AssertionError):
to64(-1)
with pytest.raises(AssertionError):
to64('a')