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'