Adding code coverage

[john-science]

What is the change?

Adding code coverage

Why is the change being made?

Trying to fight for 90%.

---

Checklist

• This PR has only one purpose or idea.
• Tests have been added/updated to verify any new/changed code.

• The code style follows good practices.
• The commit message(s) follow good practices.

• The release notes have been updated if necessary.
• The documentation is still up-to-date in the doc folder.
• The dependencies are still up-to-date in pyproject.toml.

[opotowsky]

Seems like you'll have to find some test crumbs! The code changes look fine. I'll wait for the cleanup commit to approve

[john-science]

Ah! So close.

This PR takes the code coverage from 89.56 to 89.91.

Soooooo close!