Added assert in test cases