add generator test