from unittest import TestCase
from benchmark.caesar_cipher import encrypt
from benchmark.caesar_cipher import decrypt


class Test_encrypt(TestCase):
    #  distances_true = {1: [3, 47, 0, 0, 46, 0, 22]}
    #  distances_false = {1: [0, 0, 31, 19, 0, 8, 0]}
    def test_encrypt_1(self):
        assert encrypt(strng='V*wk+`C', key=38) == "|P>2Q'i"


class Test_decrypt(TestCase):
    #  distances_true = {2: [0, 215, 203, 206, 217, 13]}
    #  distances_false = {2: [21, 0, 0, 0, 0, 0]}
    def test_decrypt_1(self):
        assert decrypt(strng='C."%0d', key=56) == 'jöêíø,'