Generating the test dataset