tests: Add a couple of code generation tests