Can you add a test case for this function?