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) == '('