Use TestNG assertions?