Using testng assertions