add a new test.py