diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/ec/test_context.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/test/ec/test_context.py b/test/ec/test_context.py index 9ad8b52..f76995b 100644 --- a/test/ec/test_context.py +++ b/test/ec/test_context.py @@ -89,6 +89,38 @@ def test_default_no_enter(): with local(DefaultContext()) as default, pytest.raises(ValueError): default.exit_action(RandomModAction(7)) +def test_multiple_enter(mult): + default = DefaultContext() + with local(default) as ctx1: + mult.multiply(59) + + with local(default) as ctx2: + mult.multiply(135) + + assert len(default.actions) == 0 + assert len(ctx1.actions) == len(ctx2.actions) + +def test_multiple_enter_chained(mult): + default = DefaultContext() + with local(default) as ctx1: + mult.multiply(59) + + with local(ctx1) as ctx2: + mult.multiply(135) + + assert len(default.actions) == 0 + assert 2 * len(ctx1.actions) == len(ctx2.actions) + +def test_multiple_enter_no_copy(mult): + default = DefaultContext() + with local(default, copy=False) as ctx1: + mult.multiply(59) + + with local(default, copy=False) as ctx2: + mult.multiply(135) + + assert len(default.actions) == len(ctx1.actions) + assert len(ctx1.actions) == len(ctx2.actions) def test_path(mult, secp128r1): with local(PathContext([0, 1])) as ctx: |