Create test.py

source code