Adapt generate_data to existing data in test

Some datasets are not coming from test, this needs to be changed. We must refactor the whole generate data, and all examples using currently iav.