"""Tests for the Caeser module"""

import pycipher

from lantern.modules import atbash


def _test_atbash(plaintext, *fitness_functions, top_n=1):
    ciphertext = pycipher.Atbash().encipher(plaintext, keep_punct=True)
    decryption = atbash.decrypt(ciphertext)

    assert decryption == plaintext.upper()


def test_decrypt():
    """Test decryption"""
    assert atbash.decrypt("uozt{Yzybolm}") == "flag{Babylon}"


def test_encrypt():
    """Test encrypt"""
    assert ''.join(atbash.encrypt("flag{Babylon}")) == "uozt{Yzybolm}"

"""Tests for the Caeser module"""

from lantern.modules import atbash


def test_decrypt():
    """Test decryption"""
    assert atbash.decrypt("uozt{Yzybolm}") == "flag{Babylon}"


def test_encrypt():
    """Test encryption"""
    assert ''.join(atbash.encrypt("flag{Babylon}")) == "uozt{Yzybolm}"

Remove unnecessary testing code from atbash
Remove unnecessary testing code from atbash
