Extra tests

[vapavlo]

Checklist

• My pull request has a clear and explanatory title.
• My pull request passes the Linting test.
• I added appropriate unit tests and I made sure the code passes all unit tests. (refer to comment below)
• My PR follows PEP8 guidelines. (refer to comment below)
• My code is properly documented, using numpy docs conventions, and I made sure the documentation renders properly.
• I linked to issues and PRs that are relevant to this PR.

Description

Issue

Additional context

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 90.96%. Comparing base (04f9665) to head (38fbf32).

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #129      +/-   ##
==========================================
+ Coverage   89.51%   90.96%   +1.44%     
==========================================
  Files         126      126              
  Lines        3518     3518              
==========================================
+ Hits         3149     3200      +51     
+ Misses        369      318      -51     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.