Update test.py
woo