mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-27 15:01:08 +00:00
4cddb26908
* atbash.py: Tighten up the benchmarks
* updating DIRECTORY.md
Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
""" https://en.wikipedia.org/wiki/Atbash """
|
|
import string
|
|
|
|
|
|
def atbash_slow(sequence: str) -> str:
|
|
"""
|
|
>>> atbash_slow("ABCDEFG")
|
|
'ZYXWVUT'
|
|
|
|
>>> atbash_slow("aW;;123BX")
|
|
'zD;;123YC'
|
|
"""
|
|
output = ""
|
|
for i in sequence:
|
|
extract = ord(i)
|
|
if 65 <= extract <= 90:
|
|
output += chr(155 - extract)
|
|
elif 97 <= extract <= 122:
|
|
output += chr(219 - extract)
|
|
else:
|
|
output += i
|
|
return output
|
|
|
|
|
|
def atbash(sequence: str) -> str:
|
|
"""
|
|
>>> atbash("ABCDEFG")
|
|
'ZYXWVUT'
|
|
|
|
>>> atbash("aW;;123BX")
|
|
'zD;;123YC'
|
|
"""
|
|
letters = string.ascii_letters
|
|
letters_reversed = string.ascii_lowercase[::-1] + string.ascii_uppercase[::-1]
|
|
return "".join(
|
|
letters_reversed[letters.index(c)] if c in letters else c for c in sequence
|
|
)
|
|
|
|
|
|
def benchmark() -> None:
|
|
"""Let's benchmark our functions side-by-side..."""
|
|
from timeit import timeit
|
|
|
|
print("Running performance benchmarks...")
|
|
setup = "from string import printable ; from __main__ import atbash, atbash_slow"
|
|
print(f"> atbash_slow(): {timeit('atbash_slow(printable)', setup=setup)} seconds")
|
|
print(f"> atbash(): {timeit('atbash(printable)', setup=setup)} seconds")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
for example in ("ABCDEFGH", "123GGjj", "testStringtest", "with space"):
|
|
print(f"{example} encrypted in atbash: {atbash(example)}")
|
|
benchmark()
|