Added more testing for generating test code