Update test.py

update this files...