def test_unwrap(self, alg, expected, hex_kek, hex_wrapped):
bin_kek = unhexlify(hex_kek)
bin_wrapped = unhexlify(hex_wrapped)
aes_key = CryptographyAESKey(bin_kek, alg)
actual = hexlify(aes_key.unwrap_key(bin_wrapped)).upper()
assert actual == expected
def test_wrap(self, alg, hex_key, hex_kek, expected):
bin_key = unhexlify(hex_key)
bin_kek = unhexlify(hex_kek)
aes_key = CryptographyAESKey(bin_kek, alg)
actual = hexlify(aes_key.wrap_key(bin_key)).upper()
assert actual == expected