Added test cases

This commit is contained in:
Harshil Darji
2016-08-02 23:16:55 +05:30
parent 525b945446
commit f68690f296
5 changed files with 127 additions and 47 deletions

View File

@ -28,9 +28,17 @@ def checkValidKey(key):
sys.exit('Error in the key or symbol set.')
def encryptMessage(key, message):
"""
>>> encryptMessage('LFWOAYUISVKMNXPBDCRJTQEGHZ', 'Harshil Darji')
'Ilcrism Olcvs'
"""
return translateMessage(key, message, 'encrypt')
def decryptMessage(key, message):
"""
>>> decryptMessage('LFWOAYUISVKMNXPBDCRJTQEGHZ', 'Ilcrism Olcvs')
'Harshil Darji'
"""
return translateMessage(key, message, 'decrypt')
def translateMessage(key, message, mode):