Improving test assertions