Improve unit testing

[sashirestela]

No description provided.

[codecov-commenter]

⚠️ Please install the to ensure uploads and comments are reliably processed by Codecov.

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 89.04%. Comparing base (63cc469) to head (dc7d663).
Report is 1 commits behind head on main.

❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

@@            Coverage Diff            @@
##               main      #10   +/-   ##
=========================================
  Coverage     89.04%   89.04%           
  Complexity       23       23           
=========================================
  Files             3        3           
  Lines            73       73           
  Branches          4        4           
=========================================
  Hits             65       65           
  Misses            7        7           
  Partials          1        1           

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