Added unit test for wrapper functions