Write Unit Tests

@hamzafer/write-unit-tests

Create comprehensive unit tests for the current code and generate the test file with proper imports and setup according to the project's testing conventions.