adding generator unit tests