The economic outlook and government books are in slightly better than expected shape, but still face a challenging couple of years, according to pre-election forecasts issued by the Treasury.