Adding assertions to the test case