Create test.py

create test.py