create test.py

create test.py