Add tests to data_test.go