add test for generators