from unittest import TestCase from benchmark.caesar_cipher import encrypt class Test_encrypt(TestCase): def test_encrypt_1(self): assert encrypt(strng='.1zZ`MJ$', key=55) == 'ehR28%"[' def test_encrypt_2(self): assert encrypt(strng='Fs;g{w', key=4) == 'Jw?k {' def test_encrypt_3(self): assert encrypt(strng='X"N{', key=49) == '*S M' def test_encrypt_4(self): assert encrypt(strng=',u', key=45) == 'YC' def test_encrypt_5(self): assert encrypt(strng='XR&lKm.', key=64) == '93fM,Nn' def test_encrypt_6(self): assert encrypt(strng='Ocj=e', key=49) == '!5