From 28390ec1575e0af026be2bfea6fd0bca8f55c008 Mon Sep 17 00:00:00 2001
From: J08nY
Date: Tue, 23 Apr 2019 19:05:27 +0200
Subject: Add proper context management.

---
 test/ec/test_context.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 test/ec/test_context.py

(limited to 'test')

diff --git a/test/ec/test_context.py b/test/ec/test_context.py
new file mode 100644
index 0000000..51fefd7
--- /dev/null
+++ b/test/ec/test_context.py
@@ -0,0 +1,23 @@
+from unittest import TestCase
+
+from pyecsca.ec.context import local, DefaultContext
+from pyecsca.ec.mult import LTRMultiplier
+from test.ec.curves import get_secp128r1
+
+
+class ContextTests(TestCase):
+
+    def setUp(self):
+        self.secp128r1 = get_secp128r1()
+        self.base = self.secp128r1.generator
+        self.coords = self.secp128r1.curve.coordinate_model
+        self.mult = LTRMultiplier(self.secp128r1, self.coords.formulas["add-1998-cmo"],
+                                  self.coords.formulas["dbl-1998-cmo"], self.coords.formulas["z"])
+
+    def test_null(self):
+        self.mult.multiply(59, self.base)
+
+    def test_default(self):
+        with local(DefaultContext()) as ctx:
+            self.mult.multiply(59, self.base)
+        self.assertEqual(len(ctx.actions), 10)
-- 
cgit v1.2.3-70-g09d2