Add data unit tests