def template_test_one_key(self, n, size):
"""n is the number of iterations.
size is the size of the message."""
for _ in range(n):
m = generate_random_string(size)
key = randint(1, MAX_MAPPED_INT - find_max_char_ord_value(m))
cipher = encrypt(m, key)
o = decrypt(cipher, key)
self.assertEqual(m, o)
def test_empty_message(self):
for i in range(100):
m = ""
cipher = encrypt(m, i)
o = decrypt(cipher, i)
self.assertEqual(m, o)