diff options
Diffstat (limited to 'test/ec/test_context.py')
| -rw-r--r-- | test/ec/test_context.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/test/ec/test_context.py b/test/ec/test_context.py index 9e56afb..1073fc0 100644 --- a/test/ec/test_context.py +++ b/test/ec/test_context.py @@ -1,7 +1,9 @@ import ast from unittest import TestCase -from pyecsca.ec.context import local, DefaultContext, OpResult, NullContext, getcontext +from pyecsca.ec.context import (local, DefaultContext, OpResult, NullContext, getcontext, + setcontext, + resetcontext) from pyecsca.ec.coordinates import AffineCoordinateModel from pyecsca.ec.mod import Mod from pyecsca.ec.mult import LTRMultiplier @@ -33,9 +35,13 @@ class ContextTests(TestCase): 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.base) self.assertEqual(len(ctx.actions), 10) + self.assertEqual(len(getcontext().actions), 0) def test_execute(self): with self.assertRaises(ValueError): |