Source code for pycrypt.translators.test_substitutiontranslator

import unittest
from substitutiontranslator import *
from .. import utils

[docs]class TestSubstitutionTranslator(unittest.TestCase):
[docs]    def setUp(self):
        self.ST = SubstitutionTranslator()
        key = dict(zip(utils.alphabet, utils.alphabet))
        key['A'], key['B'], key['C'] = "C", "A", "B"
        self.ST.setKey(key)

[docs]    def test_translate(self):
        self.assertEqual(self.ST.translate("ab c"), "CA B")

[docs]    def test_encode(self):
        self.assertEqual(self.ST.encode("abc"), "BCA")

[docs]    def test_parseInput(self):
        self.assertEqual(self.ST.parseInput("abc"), "ABC")

[docs]    def test_setKey(self):
        self.assertEqual(self.ST.key.keys(), list(utils.alphabet))

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