Add unit tests to improve test coverage #63
Description
Increase the number of unit tests to improve overall test coverage for the repository. Review the current test suite, identify untested modules or functions, and add comprehensive unit tests where applicable. Ensure all new tests follow project conventions and run successfully with the CI pipeline.
NOTE: Wait for #55 to be closed. This issue is essentially filling in the gaps left from these issues.