14 lines
406 B
Python
14 lines
406 B
Python
|
from unittest import TestCase
|
||
|
|
|
||
|
|
from benchmark.railfence_cipher import railencrypt
|
||
|
|
|
||
|
|
|
||
|
|
class Test_railencrypt(TestCase):
|
||
|
|
def test_railencrypt_1(self):
|
||
|
|
assert railencrypt(st=')x8ro;BVm', k=865) == ')x8ro;BVm'
|
||
|
|
|
||
|
|
def test_railencrypt_2(self):
|
||
|
|
assert railencrypt(st='q338K a{.', k=4) == 'qa3 {3K.8'
|
||
|
|
|
||
|
|
def test_railencrypt_3(self):
|
||
|
|
assert railencrypt(st='X61*8p', k=5) == 'X61*p8'
|