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'