Update test.py

just for test