Create test.py