algorithm/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-07-05 01:09:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added test cases

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

10
ciphers/transposition_cipher.py
View File

@ -14,6 +14,10 @@ def main():
print('Output:\n%s' %(text + '|'))
def encryptMessage(key, message):
"""
>>> encryptMessage(6, 'Harshil Darji')
'Hlia rDsahrij'
"""
cipherText = [''] * key
for col in range(key):
pointer = col
@ -23,6 +27,10 @@ def encryptMessage(key, message):
return ''.join(cipherText)
def decryptMessage(key, message):
"""
>>> decryptMessage(6, 'Hlia rDsahrij')
'Harshil Darji'
"""
numCols = math.ceil(len(message) / key)
numRows = key
numShadedBoxes = (numCols * numRows) - len(message)
@ -40,4 +48,6 @@ def decryptMessage(key, message):
return "".join(plainText)
if __name__ == '__main__':
import doctest
doctest.testmod()
main()