from unittest import TestCase
from benchmark.railfence_cipher import railencrypt
class Test_railencrypt(TestCase):
def test_railencrypt_1(self):
assert railencrypt(st='6u#%', k=3) == '6u%#'
def test_railencrypt_2(self):
assert railencrypt(st='6uD*z(', k=3) == '6zu*(D'
def test_railencrypt_3(self):
assert railencrypt(st='(', k=307) == '('
