edit test.py