[FEAT] Create Unit tests for functions

[Seyi007]

Unit tests will help ensure that each function behaves as expected and facilitate easier debugging in the future.

Requirements:

• Each test case should cover normal cases, edge cases, and potential error scenarios.
• Ensure that tests are independent and can be run in isolation.
• Add assertions to check for expected outputs and error handling.
• Document any assumptions made while creating the tests.

Acceptance Criteria:

• Unit tests created for all functions.
• Tests pass without errors when run in the testing environment.
• Documentation Update with new tests