Как написать генератор на python

Одна из задач на собеседование:
«Написать функцию генератор значений от 0 до 2»

def generator():
    num = 0
    while num <= 2:
        yield num
        num += 1


mygen = generator()
for i in mygen:
    print i

Yield это ключевое слово, которое используется примерно как return — отличие в том, что функция вернёт генератор.