edited test.py