minor refactoring

open-quantum-safe · Jul 14, 2020 · 8af801e · 8af801e
1 parent 6cc2214
commit 8af801e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_kem.py b/tests/test_kem.py
@@ -35,7 +35,7 @@ def check_wrong_ciphertext(alg_name):
     with oqs.KeyEncapsulation(alg_name) as kem:
         public_key = kem.generate_keypair()
         ciphertext, shared_secret_server = kem.encap_secret(public_key)
-        wrong_ciphertext = bytes(random.getrandbits(8) for _ in range(kem.details['length_ciphertext']))
+        wrong_ciphertext = bytes(random.getrandbits(8) for _ in range(len(ciphertext)))
         shared_secret_client = kem.decap_secret(wrong_ciphertext)
         assert shared_secret_client != shared_secret_server
 

diff --git a/tests/test_sig.py b/tests/test_sig.py
@@ -36,7 +36,7 @@ def check_wrong_message(alg_name):
         message = bytes(random.getrandbits(8) for _ in range(100))
         public_key = sig.generate_keypair()
         signature = sig.sign(message)
-        wrong_message = bytes(random.getrandbits(8) for _ in range(100))
+        wrong_message = bytes(random.getrandbits(8) for _ in range(len(message)))
         assert not (sig.verify(wrong_message, signature, public_key))
 
 
@@ -52,7 +52,7 @@ def check_wrong_signature(alg_name):
         message = bytes(random.getrandbits(8) for _ in range(100))
         public_key = sig.generate_keypair()
         signature = sig.sign(message)
-        wrong_signature = bytes(random.getrandbits(8) for _ in range(sig.details['length_signature']))
+        wrong_signature = bytes(random.getrandbits(8) for _ in range(len(signature)))
         assert not (sig.verify(message, wrong_signature, public_key))
 
 
@@ -68,7 +68,7 @@ def check_wrong_public_key(alg_name):
         message = bytes(random.getrandbits(8) for _ in range(100))
         public_key = sig.generate_keypair()
         signature = sig.sign(message)
-        wrong_public_key = bytes(random.getrandbits(8) for _ in range(sig.details['length_public_key']))
+        wrong_public_key = bytes(random.getrandbits(8) for _ in range(len(public_key)))
         assert not (sig.verify(message, signature, wrong_public_key))