Creating test suite for Generator Class