def test_AES256_prf(self):
# AES256 prf
kb = h('0D674DD0F9A6806525A4D92E828BD15A')
k = string_to_key(Enctype.AES256, b'key2', b'key2')
self.assertEqual(prf(k, b'\x02\x62'), kb)
def test_AES128_prf(self):
# AES128 prf
kb = h('77B39A37A868920F2A51F9DD150C5717')
k = string_to_key(Enctype.AES128, b'key1', b'key1')
self.assertEqual(prf(k, b'\x01\x61'), kb)