Inline code generator into the unit test