Update test.py

Test