Added assertions for test cases