modify test1.py