diff options
Diffstat (limited to 'test/ec/test_formula.py')
| -rw-r--r-- | test/ec/test_formula.py | 52 |
1 files changed, 40 insertions, 12 deletions
diff --git a/test/ec/test_formula.py b/test/ec/test_formula.py index ec585ad..ffae4c4 100644 --- a/test/ec/test_formula.py +++ b/test/ec/test_formula.py @@ -10,20 +10,25 @@ from pyecsca.ec.point import Point class FormulaTests(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.mdbl = self.secp128r1.curve.coordinate_model.formulas["mdbl-2007-bl"] self.jac_secp128r1 = get_params("secg", "secp128r1", "jacobian") - self.jac_dbl = self.jac_secp128r1.curve.coordinate_model.formulas["dbl-1998-hnm"] + self.jac_dbl = self.jac_secp128r1.curve.coordinate_model.formulas[ + "dbl-1998-hnm" + ] def test_wrong_call(self): with self.assertRaises(ValueError): self.add(self.secp128r1.curve.prime) with self.assertRaises(ValueError): - self.add(self.secp128r1.curve.prime, self.secp128r1.generator.to_affine(), self.secp128r1.generator.to_affine()) + self.add( + self.secp128r1.curve.prime, + self.secp128r1.generator.to_affine(), + self.secp128r1.generator.to_affine(), + ) def test_indices(self): self.assertEqual(self.add.input_index, 1) @@ -47,33 +52,56 @@ class FormulaTests(TestCase): self.assertEqual(self.add.num_addsubs, 10) def test_assumptions(self): - res = self.mdbl(self.secp128r1.curve.prime, self.secp128r1.generator, **self.secp128r1.curve.parameters) + res = self.mdbl( + self.secp128r1.curve.prime, + self.secp128r1.generator, + **self.secp128r1.curve.parameters + ) self.assertIsNotNone(res) - coords = {name: value * 5 for name, value in self.secp128r1.generator.coords.items()} + coords = { + name: value * 5 for name, value in self.secp128r1.generator.coords.items() + } other = Point(self.secp128r1.generator.coordinate_model, **coords) with self.assertRaises(UnsatisfiedAssumptionError): - self.mdbl(self.secp128r1.curve.prime, other, **self.secp128r1.curve.parameters) + self.mdbl( + self.secp128r1.curve.prime, other, **self.secp128r1.curve.parameters + ) with TemporaryConfig() as cfg: cfg.ec.unsatisfied_formula_assumption_action = "ignore" - pt = self.mdbl(self.secp128r1.curve.prime, other, **self.secp128r1.curve.parameters) + pt = self.mdbl( + self.secp128r1.curve.prime, other, **self.secp128r1.curve.parameters + ) self.assertIsNotNone(pt) def test_parameters(self): - res = self.jac_dbl(self.secp128r1.curve.prime, self.jac_secp128r1.generator, **self.jac_secp128r1.curve.parameters) + res = self.jac_dbl( + self.secp128r1.curve.prime, + self.jac_secp128r1.generator, + **self.jac_secp128r1.curve.parameters + ) self.assertIsNotNone(res) def test_symbolic(self): p = self.secp128r1.curve.prime k = FF(p) coords = self.secp128r1.curve.coordinate_model - sympy_params = {key: SymbolicMod(k(int(value)), p) for key, value in self.secp128r1.curve.parameters.items()} - symbolic_point = Point(coords, **{key: SymbolicMod(symbols(key), p) for key in coords.variables}) + sympy_params = { + key: SymbolicMod(k(int(value)), p) + for key, value in self.secp128r1.curve.parameters.items() + } + symbolic_point = Point( + coords, **{key: SymbolicMod(symbols(key), p) for key in coords.variables} + ) symbolic_double = self.dbl(p, symbolic_point, **sympy_params)[0] - generator_double = self.dbl(p, self.secp128r1.generator, **self.secp128r1.curve.parameters)[0] + generator_double = self.dbl( + p, self.secp128r1.generator, **self.secp128r1.curve.parameters + )[0] for outer_var in coords.variables: symbolic_val = getattr(symbolic_double, outer_var).x generator_val = getattr(generator_double, outer_var).x for inner_var in coords.variables: - symbolic_val = symbolic_val.subs(inner_var, k(getattr(self.secp128r1.generator, inner_var).x)) + symbolic_val = symbolic_val.subs( + inner_var, k(getattr(self.secp128r1.generator, inner_var).x) + ) self.assertEqual(Mod(int(symbolic_val), p), Mod(generator_val, p)) |