add test case add test case small change Co-authored-by: jie.wang <38901892+jievince@users.noreply.github.com>