class TestFileEncryption(unittest.TestCase):
def setUp(self):
self.aes = AES(b'z' * 16)
self.iv = b'\x01' * 16
def test_small_file_enctime(self):
filename = "small.txt"
start = time.time()
self.aes.encrypt_file(filename, iv=self.iv)
print(filename + "encrypted within %s seconds" % (time.time() - start))
def test_small_file_dectime(self):
filename = "small.txt.enc"
start = time.time()
self.aes.decrypt_file(filename)
print(filename + "decrypted within %s seconds" % (time.time() - start))
def test_2mbfile_enctime(self):
filename = "lorem-ipsum.txt"
start = time.time()
self.aes.encrypt_file(filename, iv=self.iv)
print(filename + " encrypted within %s seconds" %
(time.time() - start))
def test_2mbfile_dectime(self):
filename = "lorem-ipsum.txt.enc"
start = time.time()
self.aes.decrypt_file(filename)
print(filename + " decrypted within %s seconds" %
(time.time() - start))
