def test_vigenere_bad_key_symbol(self):
with self.assertRaises(ValueError):
etao.VigenereCipher('LEMON!').encrypt('ATTACKATDAWN')
def test_vigenere_decrypt_mixcase(self):
self.assertEqual(
etao.VigenereCipher('leMoN').decrypt('lxFoPVefRNhR'),
'ATTACKATDAWN')
def test_vigenere_decrypt_with_symbols(self):
self.assertEqual(
etao.VigenereCipher('LEMON').decrypt('LXFOPV EF RNHR!'),
'ATTACK AT DAWN!')
def test_vigenere_encrypt_mixcase(self):
self.assertEqual(
etao.VigenereCipher('lEmOn').encrypt('AtTackaTdAWn'),
'LXFOPVEFRNHR')
def test_vigenere_decrypt(self):
self.assertEqual(
etao.VigenereCipher('LEMON').decrypt('LXFOPVEFRNHR'),
'ATTACKATDAWN')