add data generation code for test cases