Adding tests to generator file