Update hill_cipher.py

Author: lighting9999

ciphers/hill_cipher.py

@@ -19,10 +19,7 @@
 """
 
 import string
-from typing import Optional
-
 import numpy as np
-
 from maths.greatest_common_divisor import greatest_common_divisor
 
 
@@ -136,25 +133,27 @@ def check_determinant(self) -> None:
             >>> HillCipher(invalid_key)  # Determinant 0
             Traceback (most recent call last):
             ...
-            ValueError: determinant modular 36 of encryption key(0) is not co prime w.r.t 36.
-            Try another key.
+            ValueError: determinant modular 36 of encryption key(0) is not co prime
+            w.r.t 36. Try another key.
 
             >>> invalid_key2 = np.array([[4, 2], [6, 3]])
             >>> HillCipher(invalid_key2)  # Determinant 0
             Traceback (most recent call last):
             ...
-            ValueError: determinant modular 36 of encryption key(0) is not co prime w.r.t 36.
-            Try another key.
+            ValueError: determinant modular 36 of encryption key(0) is not co prime
+            w.r.t 36. Try another key.
         """
-        det = int(round(np.linalg.det(self.encrypt_key)))
+        det = round(np.linalg.det(self.encrypt_key))
+        det = int(det)  # Convert to int after rounding float
+
         if det < 0:
             det = det % len(self.key_string)
 
         req_l = len(self.key_string)
         if greatest_common_divisor(det, req_l) != 1:
             msg = (
-                f"determinant modular {req_l} of encryption key({det}) "
-                f"is not co prime w.r.t {req_l}.\nTry another key."
+                f"determinant modular {req_l} of encryption key({det}) is not co prime "
+                f"w.r.t {req_l}.\nTry another key."
             )
             raise ValueError(msg)
 
@@ -184,15 +183,15 @@ def process_text(self, text: str) -> str:
             'ABCC'
         """
         chars = [char for char in text.upper() if char in self.key_string]
-
+        
         # Handle empty input case
         if not chars:
             return ""
-
+            
         last = chars[-1]
         while len(chars) % self.break_key != 0:
             chars.append(last)
-
+            
         return "".join(chars)
 
     def encrypt(self, text: str) -> str:
@@ -222,21 +221,21 @@ def encrypt(self, text: str) -> str:
         text = self.process_text(text.upper())
         if not text:
             return ""
-
+            
         encrypted = ""
 
         for i in range(0, len(text) - self.break_key + 1, self.break_key):
             # Extract batch of characters
             batch = text[i : i + self.break_key]
-
+            
             # Convert to numerical vector
             vec = [self.replace_letters(char) for char in batch]
             batch_vec = np.array([vec]).T
-
+            
             # Matrix multiplication and mod 36
             product = self.encrypt_key.dot(batch_vec)
             batch_encrypted = self.modulus(product).T.tolist()[0]
-
+            
             # Convert back to characters
             encrypted_batch = "".join(
                 self.replace_digits(num) for num in batch_encrypted
@@ -245,6 +244,7 @@ def encrypt(self, text: str) -> str:
 
         return encrypted
 
+
     def make_decrypt_key(self) -> np.ndarray:
         """
         Compute decryption key matrix from encryption key.
@@ -261,20 +261,22 @@ def make_decrypt_key(self) -> np.ndarray:
             >>> cipher.make_decrypt_key()
             array([[ 6, 25],
                    [ 5, 26]])
-
+            
             >>> key3x3 = np.array([[1,2,3],[4,5,6],[7,8,9]])
             >>> cipher3 = HillCipher(key3x3)
             >>> cipher3.make_decrypt_key()  # Determinant 0 should be invalid
             Traceback (most recent call last):
             ...
-            ValueError: determinant modular 36 of encryption key(0) is not co prime w.r.t 36.
-            Try another key.
+            ValueError: determinant modular 36 of encryption key(0) is not co prime
+            w.r.t 36. Try another key.
         """
-        det = int(round(np.linalg.det(self.encrypt_key)))
+        det = round(np.linalg.det(self.encrypt_key))
+        det = int(det)  # Convert to int after rounding float
+
         if det < 0:
             det = det % len(self.key_string)
-
-        det_inv: Optional[int] = None
+        
+        det_inv: int | None = None
         for i in range(len(self.key_string)):
             if (det * i) % len(self.key_string) == 1:
                 det_inv = i
@@ -316,22 +318,22 @@ def decrypt(self, text: str) -> str:
         text = self.process_text(text.upper())
         if not text:
             return ""
-
+            
         decrypt_key = self.make_decrypt_key()
         decrypted = ""
 
         for i in range(0, len(text) - self.break_key + 1, self.break_key):
             # Extract batch of characters
             batch = text[i : i + self.break_key]
-
+            
             # Convert to numerical vector
             vec = [self.replace_letters(char) for char in batch]
             batch_vec = np.array([vec]).T
-
+            
             # Matrix multiplication and mod 36
             product = decrypt_key.dot(batch_vec)
             batch_decrypted = self.modulus(product).T.tolist()[0]
-
+            
             # Convert back to characters
             decrypted_batch = "".join(
                 self.replace_digits(num) for num in batch_decrypted
@@ -344,7 +346,7 @@ def decrypt(self, text: str) -> str:
 def main() -> None:
     """
     Command-line interface for Hill Cipher operations.
-
+    
     Steps:
     1. User inputs encryption key size
     2. User inputs encryption key matrix rows
@@ -357,14 +359,14 @@ def main() -> None:
 
     print("Enter each row of the encryption key with space separated integers")
     for i in range(n):
-        row = [int(x) for x in input(f"Row {i + 1}: ").split()]
+        row = [int(x) for x in input(f"Row {i+1}: ").split()]
         hill_matrix.append(row)
 
     hc = HillCipher(np.array(hill_matrix))
 
     print("\nWould you like to encrypt or decrypt some text?")
     option = input("1. Encrypt\n2. Decrypt\nEnter choice (1/2): ")
-
+    
     if option == "1":
         text = input("\nEnter text to encrypt: ")
         print("\nEncrypted text:")
@@ -379,21 +381,20 @@ def main() -> None:
 
 if __name__ == "__main__":
     import doctest
-
     doctest.testmod()
-
+    
     print("\nRunning sample tests...")
     key = np.array([[2, 5], [1, 6]])
     cipher = HillCipher(key)
-
+    
     # Test encryption/decryption round trip
     plaintext = "HELLO123"
     encrypted = cipher.encrypt(plaintext)
     decrypted = cipher.decrypt(encrypted)
-
+    
     print(f"\nOriginal text: {plaintext}")
     print(f"Encrypted text: {encrypted}")
     print(f"Decrypted text: {decrypted}")
-
+    
     # Run CLI interface
     main()