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