Create generate_test.py