Add test for helper generator