Added assert statements to tests