New Zealand elect to bat against Sri Lanka