Added some helper methods to TestDataLoader