1 files changed, 26 insertions, 0 deletions

diff --git a/test/ec/test_key_generation.py b/test/ec/test_key_generation.py
new file mode 100644
index 0000000..59f3b23
--- /dev/null
+++ b/test/ec/test_key_generation.py
@@ -0,0 +1,26 @@
+from unittest import TestCase
+
+from pyecsca.ec.curves import get_params
+from pyecsca.ec.key_generation import KeyGeneration
+from pyecsca.ec.mult import LTRMultiplier
+
+
+class KeyGenerationTests(TestCase):
+
+    def setUp(self):
+        self.secp128r1 = get_params("secg", "secp128r1", "projective")
+        self.add = self.secp128r1.curve.coordinate_model.formulas["add-2007-bl"]
+        self.dbl = self.secp128r1.curve.coordinate_model.formulas["dbl-2007-bl"]
+        self.mult = LTRMultiplier(self.add, self.dbl)
+
+    def test_basic(self):
+        generator = KeyGeneration(self.mult, self.secp128r1)
+        priv, pub = generator.generate()
+        self.assertIsNotNone(priv)
+        self.assertIsNotNone(pub)
+        self.assertTrue(self.secp128r1.curve.is_on_curve(pub))
+        generator = KeyGeneration(self.mult, self.secp128r1, True)
+        priv, pub = generator.generate()
+        self.assertIsNotNone(priv)
+        self.assertIsNotNone(pub)
+        self.assertTrue(self.secp128r1.curve.is_on_curve(pub))