Add Generator tests