17 lines
605 B
Python
17 lines
605 B
Python
from unittest import TestCase
|
|
from benchmark.caesar_cipher import encrypt
|
|
from benchmark.caesar_cipher import decrypt
|
|
|
|
|
|
class Test_encrypt(TestCase):
|
|
# distances_true = {1: [0, 0, 0, 0, 40, 0, 31, 22]}
|
|
# distances_false = {1: [33, 14, 23, 29, 0, 28, 0, 0]}
|
|
def test_encrypt_1(self):
|
|
assert encrypt(strng='vclr.q7@', key=41) == '@-6<W;`i'
|
|
|
|
|
|
class Test_decrypt(TestCase):
|
|
# distances_true = {2: [0, 0, 1, 0, 0, 3, 1, 18, 48]}
|
|
# distances_false = {2: [1, 3, 0, 15, 6, 0, 0, 0, 0]}
|
|
def test_decrypt_1(self):
|
|
assert decrypt(strng='203$-53Db', key=19) == '~| py" 1O'
|