Test x-only ECDH

stratospher · Dec 12, 2022 · fe10103 · fe10103
1 parent a9c2165
commit fe10103
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/test/functional/test_framework/ellswift.py b/test/functional/test_framework/ellswift.py
@@ -8,6 +8,7 @@
 Do not use for anything but tests."""
 
 import random
+import unittest
 
 from .key import FE, GE, SECP256K1_G
 
@@ -76,3 +77,12 @@ def ellswift_ecdh_xonly(pubkey_theirs, privkey):
     t = FE(int.from_bytes(pubkey_theirs[32:], 'big'))
     d = int.from_bytes(privkey, 'big')
     return (d * GE.lift_x(xswiftec(u, t))).x.to_bytes()
+
+class TestFrameworkEllSwift(unittest.TestCase):
+    def test_ellswift_ecdh_xonly(self):
+        for _ in range(32):
+            privkey1, encoding1 = ellswift_create()
+            privkey2, encoding2 = ellswift_create()
+            shared_secret1 = ellswift_ecdh_xonly(encoding1, privkey2)
+            shared_secret2 = ellswift_ecdh_xonly(encoding2, privkey1)
+            assert shared_secret1 == shared_secret2
diff --git a/test/functional/test_runner.py b/test/functional/test_runner.py
@@ -70,8 +70,9 @@
 TEST_FRAMEWORK_MODULES = [
     "address",
     "blocktools",
-    "muhash",
+    "ellswift",
     "key",
+    "muhash",
     "script",
     "segwit_addr",
     "util",