Magma block cipher

[pygost.git] / pygost / test_gost3412.py

diff --git a/pygost/test_gost3412.py b/pygost/test_gost3412.py
index 6ab4c90d7076de2e3cc2e1d7a31cd7737afc681d..e0b078608884b1e36cf122d550c8e72ecbc25158 100644 (file)
--- a/pygost/test_gost3412.py
+++ b/pygost/test_gost3412.py
@@ -19,6 +19,7 @@ from unittest import TestCase
 
 from pygost.gost3412 import C
 from pygost.gost3412 import GOST3412Kuznechik
+from pygost.gost3412 import GOST3412Magma
 from pygost.gost3412 import L
 from pygost.gost3412 import PI
 from pygost.utils import hexdec
@@ -121,3 +122,17 @@ class KuznechikTest(TestCase):
     def test_decrypt(self):
         ciph = GOST3412Kuznechik(self.key)
         self.assertEqual(ciph.decrypt(self.ciphertext), self.plaintext)
+
+
+class MagmaTest(TestCase):
+    key = hexdec("ffeeddccbbaa99887766554433221100f0f1f2f3f4f5f6f7f8f9fafbfcfdfeff")
+    plaintext = hexdec("fedcba9876543210")
+    ciphertext = hexdec("4ee901e5c2d8ca3d")
+
+    def test_encrypt(self):
+        ciph = GOST3412Magma(self.key)
+        self.assertEqual(ciph.encrypt(self.plaintext), self.ciphertext)
+
+    def test_decrypt(self):
+        ciph = GOST3412Magma(self.key)
+        self.assertEqual(ciph.decrypt(self.ciphertext), self.plaintext)