Update test.py

TEST