from unittest import TestCase

from benchmark.railfence_cipher import raildecrypt


class Test_raildecrypt(TestCase):
    def test_raildecrypt_1(self):
        assert raildecrypt(st='q338K a{.', k=9) == 'q338K a{.'

    def test_raildecrypt_2(self):
        assert raildecrypt(st='q338K a{.', k=3) == 'q8{K3 .a3'

    def test_raildecrypt_3(self):
        assert raildecrypt(st='Hi=BC/', k=557) == 'Hi=BC/'

    def test_raildecrypt_4(self):
        assert raildecrypt(st='X61*8p', k=5) == 'X61*p8'