piglatin.py

"""Turn a phrase into Pig Latin.

This takes a space-separated phrase and returns it in Pig Latin.

Rules:

1. If the word begins with a consonant (not a, e, i, o, u),
   move the first letter to the end and add 'ay'

2. If the word begins with a vowel, add 'yay' to the end

For example:

    >>> pig_latin('hello awesome programmer')
    'ellohay awesomeyay rogrammerpay'

"""


vowels = {"a", "e", "i", "o", "u"}

def pig_latin(phrase):
    """Turn a phrase into pig latin.

    There will be no uppercase letters or punctuation in the phrase.

        >>> pig_latin('hello awesome programmer')
        'ellohay awesomeyay rogrammerpay'
    """

    words = phrase.split(" ")
    sentence = ""
    for word in words:
        sentence += pig_latin_word(word)
        sentence += " "
    return sentence[:-1]



def pig_latin_word(word):
    if word[0] in vowels:
        return word + "yay"
    else:
        return word[1:]+ word[0] + "ay"

pig_latin('hello awesome programmer')



if __name__ == '__main__':
    import doctest
    if doctest.testmod().failed == 0:
        print "\n*** ALL TESTS PASSED. REATGAY OBJAY!\n"