Add more tests

[mostafa]

Ticket(s)

• Add more tests #321

Description

This PR includes housekeeping, dependency updates and more tests. While adding tests I also fixed a few issues I found.

Related PRs

N/A

Development Checklist

• I have added a descriptive title to this PR.
• I have squashed related commits together.
• I have rebased my branch on top of the latest main branch.
• I have performed a self-review of my own code.
• I have commented on my code, particularly in hard-to-understand areas.
• I have added docstring(s) and type annotations to my code.
• I have made corresponding changes to the documentation (docs).
• I have added tests for my changes.

Legal Checklist

• I have read the CONTRIBUTING document.
• I have read and understood the Code of Conduct.
• I have read and agreed to the Apache CLA (required).
• I have read and agreed to the LICENSE (required).