def test_ecdh(): # Try ECDH, see if we get the same answer (privA, pubA) = P256.key_pair() (privB, pubB) = P256.key_pair() Zab = P256.dh_z(privA, pubB) Zba = P256.dh_z(privB, pubA) assert Zab == Zba