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))