[General]: Test coverage #168
Description
Issue Type
Other
Description
Add more tests coverage.
Following the existing test structure and tools, use unittest and pytest to add more test cases to improve the coverage
Steps to Reproduce
No response
Additional Information
No response