implememnt identity property

2024-07-04 09:13:48 -05:00
parent e2b6ca014e
commit 5e3164feea
2 changed files with 25 additions and 2 deletions
--- a/test/test_darc.py
+++ b/test/test_darc.py
@@ -12,7 +12,7 @@ import pytest
 )
 def test_darc(height, width, message_len):
    alphabet = SubstitutionKey.init_matrix(width, height, 255)
-    medium = SubstitutionKey(matrix=[[0 for _ in range(width)] for _ in range(height)])
+    medium = SubstitutionKey.init_identity_matrix(width, height)

    server_keys, mutual_keys, client_keys = generate_keys(height, width)
    original_message = OuterKey.init_matrix(height).matrix[0]