updating test for generators

Add more tests to generator testing to be sure that it works as expected.