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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
from unittest import TestCase
from pyecsca.ec.context import (local, DefaultContext, NullContext, getcontext,
setcontext, resetcontext, Tree)
from pyecsca.ec.params import get_params
from pyecsca.ec.mod import RandomModAction
from pyecsca.ec.mult import LTRMultiplier, ScalarMultiplicationAction
class TreeTests(TestCase):
def test_walk_by_key(self):
tree = Tree()
tree["a"] = Tree()
tree["a"]["1"] = Tree()
tree["a"]["2"] = Tree()
self.assertIn("a", tree)
self.assertIsInstance(tree.get_by_key([]), Tree)
self.assertIsInstance(tree.get_by_key(["a"]), Tree)
self.assertIsInstance(tree.get_by_key(["a", "1"]), Tree)
def test_walk_by_index(self):
tree = Tree()
a = Tree()
tree["a"] = a
d = Tree()
b = Tree()
tree["a"]["d"] = d
tree["a"]["b"] = b
self.assertIn("a", tree)
with self.assertRaises(ValueError):
tree.get_by_index([])
self.assertEqual(tree.get_by_index([0]), ("a", a))
self.assertEqual(tree.get_by_index([0, 0]), ("d", d))
def test_repr(self):
tree = Tree()
tree["a"] = Tree()
tree["a"]["1"] = Tree()
tree["a"]["2"] = Tree()
txt = tree.repr()
self.assertEqual(txt.count("\t"), 2)
self.assertEqual(txt.count("\n"), 3)
class ContextTests(TestCase):
def setUp(self):
self.secp128r1 = get_params("secg", "secp128r1", "projective")
self.base = self.secp128r1.generator
self.coords = self.secp128r1.curve.coordinate_model
self.mult = LTRMultiplier(self.coords.formulas["add-1998-cmo"],
self.coords.formulas["dbl-1998-cmo"], self.coords.formulas["z"])
self.mult.init(self.secp128r1, self.base)
def test_null(self):
with local() as ctx:
self.mult.multiply(59)
self.assertIsInstance(ctx, NullContext)
def test_default(self):
token = setcontext(DefaultContext())
self.addCleanup(resetcontext, token)
with local(DefaultContext()) as ctx:
self.mult.multiply(59)
self.assertEqual(len(ctx.actions), 1)
self.assertIsInstance(next(iter(ctx.actions.keys())), ScalarMultiplicationAction)
self.assertEqual(len(getcontext().actions), 0)
def test_default_no_enter(self):
with local(DefaultContext()) as default:
with self.assertRaises(ValueError):
default.exit_action(RandomModAction(7))
def test_str(self):
with local(DefaultContext()) as default:
self.mult.multiply(59)
str(default)
str(default.actions)
with local(NullContext()) as null:
self.mult.multiply(59)
str(null)
|
