diff options
Diffstat (limited to 'test/ec/test_context.py')
| -rw-r--r-- | test/ec/test_context.py | 30 |
1 files changed, 5 insertions, 25 deletions
diff --git a/test/ec/test_context.py b/test/ec/test_context.py index 6b9c526..3f46c9d 100644 --- a/test/ec/test_context.py +++ b/test/ec/test_context.py @@ -1,22 +1,9 @@ -import ast from unittest import TestCase -from pyecsca.ec.context import (local, DefaultContext, OpResult, NullContext, getcontext, - setcontext, - resetcontext) -from pyecsca.ec.coordinates import AffineCoordinateModel +from pyecsca.ec.context import (local, DefaultContext, NullContext, getcontext, + setcontext, resetcontext) from pyecsca.ec.curves import get_params -from pyecsca.ec.mod import Mod -from pyecsca.ec.mult import LTRMultiplier -from pyecsca.ec.point import Point - - -class OpResultTests(TestCase): - - def test_str(self): - for op, char in zip((ast.Add(), ast.Sub(), ast.Mult(), ast.Div()), "+-*/"): - res = OpResult("X1", Mod(0, 5), op, Mod(2, 5), Mod(3, 5)) - self.assertEqual(str(res), "X1") +from pyecsca.ec.mult import LTRMultiplier, ScalarMultiplicationAction class ContextTests(TestCase): @@ -40,17 +27,10 @@ class ContextTests(TestCase): with local(DefaultContext()) as ctx: self.mult.multiply(59) - self.assertEqual(len(ctx.actions), 10) + self.assertEqual(len(ctx.actions), 1) + self.assertIsInstance(next(iter(ctx.actions.keys())), ScalarMultiplicationAction) self.assertEqual(len(getcontext().actions), 0) - def test_execute(self): - with self.assertRaises(ValueError): - getcontext().execute(self.coords.formulas["z"], self.base, self.base) - with self.assertRaises(ValueError): - getcontext().execute(self.coords.formulas["z"], - Point(AffineCoordinateModel(self.secp128r1.curve.model), - x=Mod(1, 5), y=Mod(2, 5))) - def test_str(self): with local(DefaultContext()) as default: self.mult.multiply(59) |