added test.py for test