27 lines
768 B
Python
27 lines
768 B
Python
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) == '~'
|