Tirsvad
621f5f7272
- Implement decrypt function that reverses encryption with negative shift - Export decrypt in package's public API - Update tests with basic roundtrip and edge case tests - Update documentation to reflect encryption/decryption support
40 lines
941 B
Python
40 lines
941 B
Python
from caesar_cipher import decrypt, encrypt
|
|
from caesar_cipher.env import load_env
|
|
|
|
|
|
def test_encrypt_basic():
|
|
assert encrypt("hello", 5) == "mjqqt"
|
|
|
|
|
|
def test_wraps_around_alphabet():
|
|
assert encrypt("xyz", 3) == "abc"
|
|
assert encrypt("abc", 29) == "def"
|
|
|
|
|
|
def test_non_letters_and_case_preserved():
|
|
assert encrypt("Hi there, 42!", 1) == "Ij uifsf, 42!"
|
|
|
|
|
|
def test_negative_shift():
|
|
assert encrypt("def", -3) == "abc"
|
|
|
|
|
|
def test_load_env(tmp_path):
|
|
f = tmp_path / ".env"
|
|
f.write_text("# c\nA=1\nB='two'\n", encoding="utf-8")
|
|
assert load_env(f) == {"A": "1", "B": "two"}
|
|
assert load_env(tmp_path / "missing") == {}
|
|
|
|
|
|
def test_decrypt_basic():
|
|
assert decrypt("mjqqt", 5) == "hello"
|
|
|
|
|
|
def test_decrypt_roundtrip():
|
|
assert decrypt(encrypt("Hello, World!", 7), 7) == "Hello, World!"
|
|
|
|
|
|
def test_decrypt_wraps_around_alphabet():
|
|
assert decrypt("abc", 3) == "xyz"
|
|
assert decrypt("def", 29) == "abc"
|