from unittest import TestCase

from benchmark.caesar_cipher import encrypt


class Test_encrypt(TestCase):
    def test_encrypt_1(self):
        assert encrypt(strng='ToU}[G4}', key=70) == ';V<dB.zd'

    def test_encrypt_2(self):
        assert encrypt(strng='Ja%b7$x?S', key=91) == 'F]!^3 t;O'