Simplify Tests #23
Description
This issue suggests simplifying all tests in the project by following best practices for testing functions and using mocks whenever possible. This approach aims to make it easier to create new tests when adding functions or expanding coverage to include more scenarios.