Update test.py