diff options
Diffstat (limited to 'test/ec/test_key_generation.py')
| -rw-r--r-- | test/ec/test_key_generation.py | 37 |
1 files changed, 18 insertions, 19 deletions
diff --git a/test/ec/test_key_generation.py b/test/ec/test_key_generation.py index 7eb26f0..512aaac 100644 --- a/test/ec/test_key_generation.py +++ b/test/ec/test_key_generation.py @@ -1,25 +1,24 @@ -from unittest import TestCase +import pytest -from pyecsca.ec.params 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) +@pytest.fixture() +def mult(secp128r1): + add = secp128r1.curve.coordinate_model.formulas["add-2007-bl"] + dbl = secp128r1.curve.coordinate_model.formulas["dbl-2007-bl"] + return LTRMultiplier(add, 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)) + +def test_basic(secp128r1, mult): + generator = KeyGeneration(mult, secp128r1) + priv, pub = generator.generate() + assert priv is not None + assert pub is not None + assert secp128r1.curve.is_on_curve(pub) + generator = KeyGeneration(mult, secp128r1, True) + priv, pub = generator.generate() + assert priv is not None + assert pub is not None + assert secp128r1.curve.is_on_curve(pub) |