Update test_data.py