Update DataGeneratorTest.cpp