Adding Test suite for Generator Class