add test.py
add test.py