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


class Test_encrypt(TestCase):
    #  distances_true = {1: [1]}
    #  distances_false = {1: [0]}
    def test_encrypt_1(self):
        assert encrypt(strng='U', key=41) == '~'

    #  distances_true = {1: [0]}
    #  distances_false = {1: [1]}
    def test_encrypt_2(self):
        assert encrypt(strng='h', key=23) == ' '


class Test_decrypt(TestCase):
    #  distances_true = {2: [1]}
    #  distances_false = {2: [0]}
    def test_decrypt_1(self):
        assert decrypt(strng='C', key=35) == ' '

    #  distances_true = {2: [0]}
    #  distances_false = {2: [1]}
    def test_decrypt_2(self):
        assert decrypt(strng='7', key=24) == '~'