Create test.py

Main case