adding test data to split_data.py