diff --git a/exercises/simple-cipher/simple_cipher_test.py b/exercises/simple-cipher/simple_cipher_test.py index b54f4e24b26..6e8839cd752 100644 --- a/exercises/simple-cipher/simple_cipher_test.py +++ b/exercises/simple-cipher/simple_cipher_test.py @@ -4,7 +4,7 @@ from simple_cipher import Cipher -# Tests adapted from `problem-specifications//canonical-data.json` @ v1.2.0 +# Tests adapted from `problem-specifications//canonical-data.json` @ v2.0.0 class SimpleCipherTest(unittest.TestCase): # Utility functions @@ -65,10 +65,14 @@ def test_can_wrap_on_decode(self): cipher = Cipher('abcdefghij') self.assertEqual(cipher.decode('zabcdefghi'), 'zzzzzzzzzz') - def test_can_handle_messages_longer_than_key(self): + def test_can_encode_messages_longer_than_key(self): cipher = Cipher('abc') self.assertEqual(cipher.encode('iamapandabear'), 'iboaqcnecbfcr') + def test_can_decode_messages_longer_than_key(self): + cipher = Cipher('abc') + self.assertEqual(cipher.decode('iboaqcnecbfcr'), 'iamapandabear') + if __name__ == '__main__': unittest.main()