Add tests to UserDataSet