Write Tests for This Code

Write comprehensive tests for this code. Code: [PASTE CODE] Test framework: [jest / pytest / go test / etc.] Include: - Happy path tests for each public function - Edge cases (empty input, null values, boundary conditions) - Error cases (what should throw/return errors) - At l