test: refactor tests using `pytest.mark.parametrize`

[XuehaiPan]

Description

Describe your changes in detail.

Motivation and Context

• [X] I have raised an issue to propose this change (required for new features and bug fixes)

Closes #54

Types of changes

What types of changes does your code introduce? Put an x in all the boxes that apply:

• [x] Bug fix (non-breaking change which fixes an issue)
• [X] New feature (non-breaking change which adds core functionality)
• [ ] New environment (non-breaking change which adds 3rd-party environment)
• [ ] Breaking change (fix or feature that would cause existing functionality to change)
• [ ] Documentation (update in the documentation)
• [ ] Example (update in the folder of example)

Checklist

Go over all the following points, and put an x in all the boxes that apply. If you are unsure about any of these, don't hesitate to ask. We are here to help!

• [X] I have read the CONTRIBUTION guide (required)
• [X] My change requires a change to the documentation.
• [X] I have updated the tests accordingly (required for a bug fix or a new feature).
• [X] I have updated the documentation accordingly.
• [X] I have reformatted the code using make format (required)
• [X] I have checked the code using make lint (required)
• [X] I have ensured make test pass. (required)