Update test.py

check