add test data for augment layer