add a test.py