Add tests for data_sample