wrote test.py