Create generate_data.py

Used to generate test data.