Write unit tests for unite function