Add back test functions for datasets