Update test.py

gg