Add testbench for codegen

Test test-codegen fails