Source code for pycrypt.translators.test_vigeneretranslator

import unittest
from vigeneretranslator import *
from .. import utils
import numpy

[docs]class TestVigenereTranslator(unittest.TestCase):
[docs]    def setUp(self):
        self.T = VigenereTranslator("ABC")

[docs]    def test_translate(self):
        self.assertEqual(self.T.translate("AAABBB"), "BCDCDE")

[docs]    def test_encode(self):
        self.assertEqual(self.T.encode("B C"), "A A")

if __name__ == '__main__':
    unittest.main()