test.py update