def test_extend_keyword(): cipher = VigenereCipher('TRAIN') extended = cipher.extend_keyword(16) assert extended == 'TRAINTRAINTRAINT'