Use better unittest assertion methods