class TestBlockEncryption(unittest.TestCase): def setUp(self): self.aes128 = AES(b'z' * 16) self.aes192 = AES(b'x' * 24) self.aes256 = AES(b'c' * 32) def test_encr_block_128(self): message = b'M' * 16 ciphertext = self.aes128.encrypt_block(message) self.assertEqual(self.aes128.decrypt_block(ciphertext), message) message = b'message 123 1234' ciphertext = self.aes128.encrypt_block(message) self.assertEqual(self.aes128.decrypt_block(ciphertext), message) def test_encr_block_192(self): message = b'\x01' * 16 ciphertext = self.aes192.encrypt_block(message) self.assertEqual(self.aes192.decrypt_block(ciphertext), message) message = b'message 123 1234' ciphertext = self.aes192.encrypt_block(message) self.assertEqual(self.aes192.decrypt_block(ciphertext), message) def test_encr_block_256(self): message = b'\x01' * 16 ciphertext = self.aes256.encrypt_block(message) self.assertEqual(self.aes256.decrypt_block(ciphertext), message) message = b'message 123 1234' ciphertext = self.aes256.encrypt_block(message) self.assertEqual(self.aes256.decrypt_block(ciphertext), message)