AB#123 create unit test

[zhiyizhu805]

Pull Request Title (Mandatory)

• Create unit test

---

Description (Mandatory)

• What does this PR do?
  Add unit test

---

Checklist (Mandatory)

Please ensure your PR meets the following requirements before submitting:

• My code follows the style guidelines of this project.
• I have performed a self-review of my own code.
• I have commented my code, particularly in hard-to-understand areas.
• I have added or updated relevant documentation (if applicable).
• I have added tests that prove my fix is effective or that my feature works.
• New and existing tests pass locally with my changes.
• I have checked for any potential breaking changes.

---

Type of Change (Mandatory)

What kind of change does this PR introduce?
(Please check the relevant options)

• 🐛 Bug fix (non-breaking change which fixes an issue)
• ✨ New feature (non-breaking change which adds functionality)
• ⚠️ Breaking change (fix or feature that would cause existing functionality to not work as expected)
• 📄 Documentation update (changes to documentation only)
• 🔧 Refactoring (no functional changes, code improvements)

---

Approval Checklist for Reviewers

Reviewers, please ensure:

• The code is easy to understand and follows the project's coding standards.
• All required documentation is updated or created.
• The change is appropriately tested and validated.
• There are no breaking changes (or they are clearly documented).