Função run adicionado.
teste.py funcionando, resolvendo o problema proposto.
Enunciado do problema inserido no começo como comentário.