Added tests for the DataGenerator Package