Tweak python test task