modify test.py