Unit Test for Generator so far