[FEATURE] Unit test improvements

[NebraskaCoder]

Describe the solution you'd like

As part of Hacktoberfest, we need to add comprehensive unit tests to ensure the stability and reliability of our codebase. This task involves writing unit tests for key components and functions in the repository.

Objectives

1. Identify key components and functions that lack unit tests.
2. Write unit tests for the identified components and functions.
3. Ensure that the new tests cover a wide range of scenarios, including edge cases.
4. Run the tests to verify they work correctly and integrate with the existing test suite.

Describe alternatives you've considered

No response

Additional Context

No response