writing a test.py