Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
* Add unit test * Fix typo Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>