test.py modificado