updated test
Merge pull request #4 from musgravejw/code_gen

updated test