def test_from_worksheet():
cipher = MixedAlphabet(key="QWERTYUIOPASDFG HJKLZXCVBNM")
ciphertext = cipher.encrypt("WELCOME TO VILLIERS PARK")
assert cipher.compare_ciphertext("CTSEGDTMLGMXOSSOTJKM QJA", ciphertext)
assert cipher.compare_plaintext("WELCOME TO VILLIERS PARK",
cipher.decrypt(ciphertext))
plaintext = cipher.decrypt("WTLLTJMLIQFMEQTKQJ")
assert cipher.compare_plaintext("BETTER THAN CAESAR", plaintext)
assert cipher.compare_ciphertext("WTLLTJMLIQFMEQTKQJ",
cipher.encrypt(plaintext))
def test_from_cryptanalysis_presentation():
cipher = MixedAlphabet(key="WEKIATBOZXMGNLCPURDJYQH_FSV")
ciphertext = cipher.encrypt("HELLO EVERYONE")
assert cipher.compare_ciphertext("OAGGCVAQARFCLA", ciphertext)
assert cipher.compare_plaintext("HELLO EVERYONE",
cipher.decrypt(ciphertext))