add test.py for testing