add generator tests