Test the entire work flow with data augmentation