def test_decrypt(self):
"""Tests encrypt function for Skytale cipher
"""
self.assertEqual(Skytale.decrypt('HENTEIDTLAEAPMRCMUAK', 5),
'HELPMEIAMUNDERATTACK')
with self.assertRaises(Skytale.SkytaleError):
Skytale.decrypt(42, 5)
with self.assertRaises(Skytale.SkytaleError):
Skytale.decrypt('Help', 'Help')
with self.assertRaises(Skytale.SkytaleError):
Skytale.decrypt('Help', 5.0)
with self.assertRaises(Skytale.SkytaleError):
Skytale.decrypt('short', 5)
def test_encrypt(self):
"""Tests encrypt function for Skytale cipher
"""
self.assertEqual(Skytale.encrypt('Help me, I am under Attack', 5),
'HENTEIDTLAEAPMRCMUAK')
with self.assertRaises(Skytale.SkytaleError):
Skytale.encrypt(42, 5)
with self.assertRaises(Skytale.SkytaleError):
Skytale.encrypt('Help', 'Help')
with self.assertRaises(Skytale.SkytaleError):
Skytale.encrypt('Help', 5.0)
with self.assertRaises(Skytale.SkytaleError):
Skytale.encrypt('short', 5)
