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