Add more tests for wrapper.