Source code for pycrypt.translators.test_semaphoretranslator

import unittest
from semaphoretranslator import *
from .. import utils
import numpy

[docs]class TestSemaphoreTranslator(unittest.TestCase):
[docs]    def setUp(self):
        self.T = SemaphoreTranslator()

[docs]    def test_translate(self):
        self.assertEqual(self.T.translate("zx  ax xq"), "A BC")

[docs]    def test_encode(self):
        self.assertEqual(self.T.encode("d e"), "wx  ex")

[docs]    def test_graphicEncode(self):
        self.assertTrue((self.T.graphicEncode("ab") == numpy.array([[0, 0, 0, 0, 0, 0], [0, 1, 0, 1, 1, 0], [1, 1, 0, 0, 1, 0]])).all())

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