Add test for type lookup generation in inference phase