Update test.py

sample code