Python2017/labs02/task10.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-


"""
Napisz funkcję pokemon_speak, która zamienia w podanym napisie co drugą literę
na wielką. Np. pokemon_speak('pokemon') powinno zwrócić 'PoKeMoN'.
"""


def pokemon_speak(text):
    for i in range(len(text)):
        if i == 0:
            mytext=''.join(['', text[i].upper()])
        elif i % 2 == 0:
            mytext = ''.join([mytext[:i], text[i].upper()])
        else:
            mytext = ''.join([mytext[:i], text[i]])
    return (mytext)


def tests(f):
    inputs = [['pokemon'], ['do not want'], ['POKEMON']]
    outputs = ['PoKeMoN', 'Do nOt wAnT', 'POKEMON']

    for input, output in zip(inputs, outputs):
        if f(*input) != output:
            return "ERROR: {}!={}".format(f(*input), output)
            break
    return "TESTS PASSED"

if __name__ == "__main__":
    print(tests(pokemon_speak))