Update test.py

updated