Unit tests

Write unit tests for the following code, covering: normal cases, edge cases, and error cases. Use [TEST FRAMEWORK] in [LANGUAGE]. Explain what each test validates. Code to test: [CODE]. Objective: good coverage without redundant tests.