1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
from ast import parse
import pytest
from pyecsca.ec.formula import OpResult
from pyecsca.ec.mod import mod
from pyecsca.ec.op import CodeOp, OpType
@pytest.mark.parametrize("name,module,result,op_type",
[("add", "x = a+b", "x = a+b", OpType.Add), ("sub", "x = a-b", "x = a-b", OpType.Sub),
("mul", "y = a*b", "y = a*b", OpType.Mult), ("div", "z = a/b", "z = a/b", OpType.Div),
("inv", "z = 1/b", "z = 1/b", OpType.Inv), ("pow", "b = a**d", "b = a^d", OpType.Pow),
("sqr", "b = a**2", "b = a^2", OpType.Sqr), ("id1", "b = 7", "b = 7", OpType.Id),
("id2", "b = a", "b = a", OpType.Id), ])
def test_str(name, module, result, op_type):
code = parse(module, mode="exec")
op = CodeOp(code)
assert str(op) == result
assert op.operator == op_type
@pytest.mark.parametrize("name,module,locals,result",
[("add", "x = a+b", {"a": mod(5, 21), "b": mod(7, 21)}, mod(12, 21)),
("sub", "x = a-b", {"a": mod(7, 21), "b": mod(5, 21)}, mod(2, 21)), ])
def test_call(name, module, locals, result):
code = parse(module, mode="exec")
op = CodeOp(code)
res = op(**locals)
assert res == result
def test_opresult_repr():
res = OpResult("a", mod(7, 11), OpType.Neg, "b")
assert repr(res) == "a = -b"
res = OpResult("a", mod(5, 7), OpType.Add, "c", 3)
assert repr(res) == "a = c+3"
res = OpResult("a", mod(3, 11), OpType.Inv, "d")
assert repr(res) == "a = 1/d"
|
