update test.py