Added test

Crypto-TII · Apr 28, 2024 · 166882c · 166882c
1 parent 734b53d
commit 166882c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/unit/cipher_test.py b/tests/unit/cipher_test.py
@@ -561,3 +561,10 @@ def test_cipher_inverse():
     ciphertext = qarmav2.evaluate([key, plaintext, tweak])
     cipher_inv = qarmav2.cipher_inverse()
     assert cipher_inv.evaluate([ciphertext, tweak, key]) == plaintext
+
+
+def test_all_sboxes_are_standard():
+    aes = AESBlockCipher(number_of_rounds=2)
+    assert aes.all_sboxes_are_standard()
+    des = DESBlockCipher(number_of_rounds = 2, number_of_sboxes=8)
+    assert not des.all_sboxes_are_standard()