add a test.py for test :)