rearrange test sets

[3r1ccc]

Signed-off-by: Eric Chang ericcc@infuseai.io

PR checklist

• [X] Ensure you have added or ran the appropriate tests for your PR.
• [X] DCO signed

What type of PR is this? Chore What this PR does / why we need it: Rearrange test set Which issue(s) this PR fixes: Rearrange test set

Special notes for your reviewer:

Does this PR introduce a user-facing change?:

NONE