add test.py
for sample