adding base for a generator test
work on generator - adding tests