add a test.py file