Nucypher
/
pyUmbral
mirror of https://github.com/nucypher/pyUmbral.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Removed random from test imports. Also some linting.

pull/40/head
jMyles 2018-01-28 10:28:31 -08:00
parent 68c9c27bbb
commit 3311bb6b8e
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_umbral.py
View File

@ -1,14 +1,12 @@
import pytest import pytest
from umbral import umbral, keys from umbral import umbral, keys
import random
# (N,threshold)
from umbral.umbral import Capsule
from umbral.point import Point
from umbral.bignum import BigNum from umbral.bignum import BigNum
from umbral.point import Point
from umbral.umbral import Capsule
parameters = [ parameters = [
# (N,threshold)
(1, 1), (1, 1),
(6, 1), (6, 1),
(6, 4), (6, 4),
@ -84,7 +82,6 @@ def test_m_of_n(N, threshold):
priv_bob = pre.gen_priv() priv_bob = pre.gen_priv()
pub_bob = pre.priv2pub(priv_bob) pub_bob = pre.priv2pub(priv_bob)
sym_key, capsule = pre._encapsulate(pub_alice) sym_key, capsule = pre._encapsulate(pub_alice)
kfrags, vkeys = pre.split_rekey(priv_alice, pub_bob, threshold, N) kfrags, vkeys = pre.split_rekey(priv_alice, pub_bob, threshold, N)
@ -101,7 +98,6 @@ def test_m_of_n(N, threshold):
# assert capsule.is_openable_by_bob() # TODO: Is it possible to check here if >= m cFrags have been attached? # assert capsule.is_openable_by_bob() # TODO: Is it possible to check here if >= m cFrags have been attached?
capsule.open(pub_bob, priv_bob, pub_alice) capsule.open(pub_bob, priv_bob, pub_alice)
sym_key_2 = pre._decapsulate_reencrypted(pub_bob, priv_bob, pub_alice, capsule) sym_key_2 = pre._decapsulate_reencrypted(pub_bob, priv_bob, pub_alice, capsule)
assert sym_key == capsule.contents assert sym_key == capsule.contents
@ -167,6 +163,7 @@ def test_capsule_serialization():
umbral.UmbralParameters().curve) umbral.UmbralParameters().curve)
assert new_capsule.original_components() == capsule.original_components() assert new_capsule.original_components() == capsule.original_components()
def test_reconstructed_capsule_serialization(): def test_reconstructed_capsule_serialization():
pre = umbral.PRE(umbral.UmbralParameters()) pre = umbral.PRE(umbral.UmbralParameters())
@ -187,8 +184,8 @@ def test_reconstructed_capsule_serialization():
assert len(rec_capsule_bytes) == 99 assert len(rec_capsule_bytes) == 99
new_rec_capsule = umbral.Capsule.from_bytes( new_rec_capsule = umbral.Capsule.from_bytes(
rec_capsule_bytes, rec_capsule_bytes,
umbral.UmbralParameters().curve, is_reconstructed=True) umbral.UmbralParameters().curve, is_reconstructed=True)
assert new_rec_capsule._point_eph_e_prime == capsule._point_eph_e_prime assert new_rec_capsule._point_eph_e_prime == capsule._point_eph_e_prime
assert new_rec_capsule._point_eph_v_prime == capsule._point_eph_v_prime assert new_rec_capsule._point_eph_v_prime == capsule._point_eph_v_prime
assert new_rec_capsule._point_noninteractive == capsule._point_noninteractive assert new_rec_capsule._point_noninteractive == capsule._point_noninteractive
@ -214,7 +211,7 @@ def test_challenge_response_serialization():
assert len(ch_resp_bytes) == (33 * 4) + (32 * 3) == 228 assert len(ch_resp_bytes) == (33 * 4) + (32 * 3) == 228
new_ch_resp = umbral.ChallengeResponse.from_bytes( new_ch_resp = umbral.ChallengeResponse.from_bytes(
ch_resp_bytes, umbral.UmbralParameters().curve) ch_resp_bytes, umbral.UmbralParameters().curve)
assert new_ch_resp.point_eph_e2 == ch_resp.point_eph_e2 assert new_ch_resp.point_eph_e2 == ch_resp.point_eph_e2
assert new_ch_resp.point_eph_v2 == ch_resp.point_eph_v2 assert new_ch_resp.point_eph_v2 == ch_resp.point_eph_v2
assert new_ch_resp.point_kfrag_commitment == ch_resp.point_kfrag_commitment assert new_ch_resp.point_kfrag_commitment == ch_resp.point_kfrag_commitment