Update test_v1.py