Added assert in the test cases