Adding Unit tests to generation code and generated code