Remove test_data because the new schema is different than test_data
contents
Add some tests