change test.py