add tests.py