Enhance the test case with more assertions