First unit tests for Generator