def test_input_1(self):
plaintext = hex_to_arr(r'00112233445566778899aabbccddeeff')
key = hex_to_arr(r'000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f')
expected_out = hex_to_arr(r'8ea2b7ca516745bfeafc49904b496089')
test_out = Core.encrypt_256(plaintext, iter_key(key, 256))
for expected, test in zip(expected_out, test_out):
self.assertEqual(expected, test)
def test_variable_key(self):
for plaintext, key, expected_ciphertext in self.tests['Variable Key']:
with self.subTest(plaintext=plaintext, key=key, expected_ciphertext=expected_ciphertext):
test_ciphertext = Core.encrypt_256(hex_to_arr(plaintext), iter_key(hex_to_arr(key), 256))
for e_item, t_item in zip(hex_to_arr(expected_ciphertext), test_ciphertext):
self.assertEqual(e_item, t_item)