Improve tests

[mtfishman]

No description provided.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 34.85%. Comparing base (d5ea09d) to head (ecb68c8).
⚠️ Report is 1 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main       #2      +/-   ##
==========================================
- Coverage   36.00%   34.85%   -1.15%     
==========================================
  Files           3        3              
  Lines         175      175              
==========================================
- Hits           63       61       -2     
- Misses        112      114       +2     

Flag	Coverage Δ
docs	0.00% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

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

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.