fix: Fixed E731 errors in tests

[ghost]

Thanks for contributing!

Before submitting your pull request please have a look at the following checklist:

• [x] ran the tests (pytest)
• [x] all style issues addressed (flake8)
• [x] your changes are covered by tests
• [x] your changes are documented, if needed

In addition, please take care to provide a proper description on what your change does, fixes or achieves when submitting the pull request.

[codecov[bot]]

Codecov Report

All modified lines are covered by tests :white_check_mark:

Comparison is base (3696d53) 96.96% compared to head (d4b5d1a) 96.96%.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #737 +/- ## ======================================= Coverage 96.96% 96.96% ======================================= Files 20 20 Lines 1547 1547 ======================================= Hits 1500 1500 Misses 47 47 ```

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