mirror of
https://github.com/bitcoin/bips.git
synced 2026-11-02 19:41:29 +00:00
Fabian Jahr
1eb496599c
Add a vendored copy of the secp256k1lab library (master branch, commit a265da139aea27386085a2a8760f8698e1bda64e) that the half-aggregation reference implementation and test vector scripts depend on.
19 lines
611 B
Python
19 lines
611 B
Python
from random import randbytes
|
|
import unittest
|
|
|
|
from secp256k1lab.ecdh import ecdh_libsecp256k1
|
|
from secp256k1lab.keys import pubkey_gen_plain
|
|
|
|
|
|
class ECDHTests(unittest.TestCase):
|
|
"""Test ECDH module."""
|
|
|
|
def test_correctness(self):
|
|
seckey_alice = randbytes(32)
|
|
pubkey_alice = pubkey_gen_plain(seckey_alice)
|
|
seckey_bob = randbytes(32)
|
|
pubkey_bob = pubkey_gen_plain(seckey_bob)
|
|
shared_secret1 = ecdh_libsecp256k1(seckey_alice, pubkey_bob)
|
|
shared_secret2 = ecdh_libsecp256k1(seckey_bob, pubkey_alice)
|
|
self.assertEqual(shared_secret1, shared_secret2)
|