add tests for wrapper