def test_decrypt_AES(cipher_text, key, iv=None, block_size=BLOCK_SIZE):
"""
Tests function decrypt_AES with valid data
"""
raised = False
plain_text = Crypto.decrypt_AES(cipher_text, key, iv, block_size)
if plain_text != 'Ala ma kota.'.encode('utf-8'):
raised = True
unittest.TestCase().assertFalse(expr = raised)
def test_decrypt_AES_inavlid_data(cipher_text, key=None, iv=None, block_size=BLOCK_SIZE):
"""
Tests function decrypt_AES with valid data
"""
raised = False
try:
plain_text = Crypto.decrypt_AES(cipher_text, key, iv, block_size)
if plain_text == "Ala ma kota.".encode('utf-8'):
raised = True
except BaseException:
raised = True
unittest.TestCase().assertTrue(expr = raised)
