Update test_problem.py

Atualizando o padrão de coding do python.