add test.py for practice