Edit Generation Tests

[atharva-2001]

:pencil: Description

Type: :vertical_traffic_light: testing

The generation tests add a huge chunk of time to the test workflow. We'd want the test workflow to finish earlier.

:pushpin: Resources

Examples, notebooks, and links to useful references.

:vertical_traffic_light: Testing

How did you test these changes?

• [ ] Testing pipeline
• [ ] Other method (describe)
• [ ] My changes can't be tested (explain why)

:ballot_box_with_check: Checklist

• [ ] I requested two reviewers for this pull request
• [ ] I updated the documentation according to my changes
• [ ] I built the documentation by applying the build_docs label

Note: If you are not allowed to perform any of these actions, ping (@) a contributor.

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 67.76%. Comparing base (e12ccf3) to head (438c1fc). Report is 5 commits behind head on master.

:exclamation: Current head 438c1fc differs from pull request most recent head 2a77edc

Please upload reports for the commit 2a77edc to get more accurate results.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #2631 +/- ## ======================================= Coverage 67.76% 67.76% ======================================= Files 171 171 Lines 14411 14411 ======================================= Hits 9765 9765 Misses 4646 4646 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.

[tardis-bot]

*beep* *bop*

Hi, human.

The docs workflow has succeeded :heavy_check_mark:

Click here to see your results.