add test.py
for sample
format code
format code