add basic testing to generator