add test.py file