Update unit test cases #378
Description
Some of the existing test cases need to be updated and new test cases need to be added for code coverage.
{{ message }}
Some of the existing test cases need to be updated and new test cases need to be added for code coverage.