fixed broken performance and analysis test

analysis/aes_comparison.py

@@ -9,7 +9,7 @@
 times_with_aes = []
 times_without_aes = []
 
-for payload_length in test_sample_sizes:
+for i, payload_length in enumerate(test_sample_sizes):
     payload = randbytes(payload_length)
 
     durations = run_with_aes(payload)
@@ -18,7 +18,7 @@
     durations = run_without_aes(payload)
     times_without_aes.append(sum(durations))
 
-    print(f"Test {payload_length} completed")
+    print(f"{(i+1) / len(test_sample_sizes) * 100:.0f}% completed")
 
 # save CSV
 with open(path.join(path.dirname(__file__), "aes_comparison.csv"), "w") as file:
@@ -31,4 +31,5 @@
 plt.plot(test_sample_sizes, times_without_aes, marker="x", label="without AES")
 plt.xlabel("Payload size, bytes")
 plt.ylabel("Encryption + decryption time, seconds")
+plt.legend(loc='upper left')
 plt.show()

perf_tests/test_encryption.py

@@ -1,6 +1,6 @@
 from random import randbytes
 from time import time
-from kyber.encryption import Encrypt, Decrypt, generate_keys
+from kyber.encryption import Encrypt, decrypt, generate_keys
 from kyber.constants import k, n, du, dv
 
 def run(payload: bytes) -> tuple[float, float, float]:
@@ -19,7 +19,7 @@ def run(payload: bytes) -> tuple[float, float, float]:
     ciphertext_chunk_size = du*k*n//8 + dv*n//8
     restored_payload = bytearray()
     for i in range(0, len(ciphertext), ciphertext_chunk_size):
-        restored_payload += Decrypt(private_key, ciphertext[i:i+ciphertext_chunk_size]).decrypt()
+        restored_payload += decrypt(private_key, ciphertext[i:i+ciphertext_chunk_size])
 
     t3 = time()