Source code for pycrypt.translators.test_caesartranslator

import unittest
from caesartranslator import *

[docs]class TestCaesarTranslator(unittest.TestCase):
[docs]    def setUp(self):
        self.CT = CaesarTranslator()

[docs]    def test_translate(self):
        self.CT.setKey(13)
        self.assertEqual(self.CT.translate("abc"), "NOP")
        self.assertEqual(self.CT.translate("ab c"), "NO P")

[docs]    def test_encode(self):
        self.CT.setKey(1)
        self.assertEqual(self.CT.encode("abc"), "ZAB")

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

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